inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

2-N

2-R 반례 궁금합니다.

227

it Ez

작성한 질문수 4

0

#include <bits/stdc++.h>
using namespace std;

int N;
map<int, vector<int>> mp;
int tmp, d, ret;

void removeAll(int key){
    // if not leaf node -> recursive remove
    if(!mp[key].empty()) {
        for(int c: mp[key]) removeAll(c);
    }
    // erase itself
    mp.erase(key);
    
}

int main(){
    cin >> N;
    
    // make graph
    for(int i = 0; i < N; i++){
        mp[i];
        cin >> tmp;
        if (tmp != -1) mp[tmp].push_back(i);
    }
    // input node to be deleted 
    cin >> d;
    removeAll(d);

    if (mp.size() == 1) ret = 1;
    else if (mp.size() == 0) ret = 0;
    else {
        // for all key in map
        for(auto it: mp) {
            int key = it.first;
            // if the remaining value empty => plus
            if (mp[key].empty()) ret++;
        }
    }

    cout << ret;
}

다음과 같이 map과 재귀를 풀어서 1068번 트리 문제를 풀었는데, 어디가 오답인지 감이 안옵니다.

c++ 코딩-테스트

답변 1

0

큰돌

안녕하세요 it님ㅎㅎ

map을 이용하셔서 잘하셨네요

다만, 저렇게 할 경우 지워진 노드의 부모 노드로부터 지우는 코드가 필요합니다.

제가 좀 다듬어 봤는데요.

이렇게 해보시겠어요?

#include <bits/stdc++.h>
using namespace std;

int N;
map<int, vector<int>> mp;
int tmp, d, ret;

void removeAll(int key){ 
    if(!mp[key].empty()) {
        for(int c: mp[key]) removeAll(c);
    } 
    mp.erase(key);
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    cin >> N;
     
    for(int i = 0; i < N; i++){
        mp[i];
        cin >> tmp;
        if (tmp != -1) mp[tmp].push_back(i);
    }
 
    cin >> d;
    removeAll(d); 
    for(auto& it: mp) {
        it.second.erase(remove(it.second.begin(), it.second.end(), d), it.second.end());
    }

    ret = 0;
    for(auto it: mp) {
        if (mp[it.first].empty()) ret++;  
    }

    cout << ret;
    return 0;
}

 

 

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.

 

채점서버 연결 관련 질문입니다

0

13

0

삼성 s직군

0

12

0

삼성 코테 없어짐

0

54

1

코딩살구클럽 가입부탁드립니다

0

36

2

코딩살구클럽 가입 요청 확인부탁드립니다

0

33

2

5-S 테스트 케이스 질문

0

37

2

코살 문제풀이 환경

0

49

2

2 - T 오큰수 문제가 있는 것 같습니다.

0

44

1

추천 추가문제들

0

44

2

프로그래머스 코테 환경 관련해서 질문드립니다.

0

49

2

해당 문제에 대한 채점이 코딩살구클럽에서 올바르게 처리되지 않습니다.

0

44

2

균형 이진 트리 설명 시 높이 숫자

0

33

2

4-H 질문드립니다.

0

37

2

1-K 질문드립니다.

0

43

2

대기업 인적성 시험 질문

0

43

2

4-C 질문드립니다

0

43

2

[수학숙제 / BOJ 2870] 채점 서버 오작동

0

40

1

코테 준비 질문

0

54

1

살구클럽가입 요청드려요

0

41

2

1-I 문제 질문

0

46

2

코딩살구클럽 가입

0

59

2

AI 코딩 도구 사용 시 학습 방법 조언

0

56

2

코딩살구클럽 오류

0

65

2

코살클 [3-F 괄호 추가하기] 프라이빗 9번 제보

0

51

1