2-R 반례 궁금합니다.
235
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번 트리 문제를 풀었는데, 어디가 오답인지 감이 안옵니다.
답변 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
67
2
코딩살구클럽 가입문의
0
69
2
3-K private case 질문
0
66
2
안녕하세요. 코딩살구클럽 문의드립니다!
0
77
2
7:41 듣다가 질문합니다
0
77
2
코드 리뷰 요청드립니다!
0
83
2
코딩살구클럽 계정 문의
0
79
2
SK하이닉스 대비 강의수강
0
217
3
코딩살구클럽 등록부탁드립니다.
0
78
2
7-F 재귀로 풀시 런타임애러
0
75
1
1-N 문제 질문입니다. (+ 채점오류)
0
82
2
코딩살구클럽 가입 확인 부탁드립니다.
0
86
2
직장인 코테 합격 공부방법 문의
0
127
2
코딩살구클럽 가입 방법
0
96
2
6-H 체점 관련 질문
0
74
1
채점서버 연결 관련 질문입니다
0
101
2
삼성 s직군
0
88
1
삼성 코테 없어짐
0
212
1
코딩살구클럽 가입부탁드립니다
0
87
2
코딩살구클럽 가입 요청 확인부탁드립니다
0
69
2
5-S 테스트 케이스 질문
0
78
2
코살 문제풀이 환경
0
88
2
2 - T 오큰수 문제가 있는 것 같습니다.
0
81
1
추천 추가문제들
0
74
2





