inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

2-N

2-R 반례 궁금합니다.

218

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

4

1

추천 문제

0

7

1

코딩살구클럽 승인

0

9

1

코살구 1주차 1940번 문제 조건과 프라이빗 테스트 불일치 문의

0

21

2

문제를 고민하는 시간 관련

0

26

2

코딩살구클럽

0

38

2

코딩살구클럽 문의

0

37

2

코딩살구클럽 승인

0

35

2

DP 경우의 수 설명이 이해가 되지 않습니다.

0

33

2

3-F 채점 관련 질문

0

31

1

BFS, DFS 활용이 되는 상황에서의 방향성

0

33

2

코딩살구클럽 승인

0

45

2

코딩살구클럽승인

0

39

3

코딩살구클럽 승인

0

54

2

3-D 관련 질문

0

35

2

코살구 회원가입 문의

0

45

2

코살구 로그인 문제

0

65

2

3-A 문제 풀이 관련 질문

0

56

3

2-O 질문 있습니다

0

38

2

2-T 문제에 관한 질문

0

40

2

코딩 살구 클럽 접속 및 사용방법 문의

0

63

2

안녕하세요~. 현재 코살코딩클럽 사이트가 접속이 안됩니다~

0

67

2

코딩살구클럽 로그인문제

0

85

3

코딩 살구 클럽 로그인 문제

0

86

2