inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

2-N

2-R 반례 궁금합니다.

215

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점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.

 

4 - A

0

10

1

코딩살구클럽 입장이 안됩니다

0

49

2

4-F 경우의 수 질문입니다.

0

30

2

코딩살구클럽 가입이 안됩니다.

0

63

2

살구 클럽에 대한 질문있습ㄴ디ㅏ

0

53

1

교안 158페이지 문의드립니다

0

44

2

코딩살구클럽 관련 건의사항

0

106

1

코살에 19942 다이어트 문제에 N의 범위가 빠져있슴니다

0

44

1

진행 방법 질문드립니다!

0

78

2

2-I) 왜 이 문제가 그래프이론 카테고리에 있는지 잘 모르겠습니다.

0

63

2

2주차 개념#12 트리 순회

0

32

2

백준사이트가 종료된다고 합니다.

0

307

2

백준 서비스 종료

9

943

1

sk 하이닉스 코테 대비

0

382

2

3-G 최댓값 질문

0

53

1

모듈러 연산 값이 10이 아닌 경우도 있지 않나요?

0

84

2

3-I 코드 질문드립니다.

0

63

2

3-N 질문 있습니다.

0

68

2

학습방법

0

105

2

4-H 질문 있습니다 (코드 리뷰)

0

68

2

코딩테스트 어디까지 준비해야 하는지 질문이 있습니다.

0

179

2

2-O 반례가 무엇일지 어떤 부분이 틀렸는지 잘 모르겠습니다.

0

71

2

2주차 개념 #4-2. 인접행렬 질문있습니다.

0

65

2

1-A 문제풀이 후 궁금한 점이 생겨서 질문드립니다.

0

53

2