이 코드에서 erase 함수는 불필요할까요?

작성한 질문수 5

0

안녕하세요 민규님 ㅎㅎ

잘 짜셨네요. ㅎㅎ

강의에서 dfs 하나만으로 푸시는걸 보니 굳이 erase가 필요할까 싶기도 했네요..

>> 필요한 문제도 있고 필요하지 않은 문제도 있습니다. 다만 erase() 의 경우 cost기 때문에 이부분을 어떻게 하면 안사용할까? 라는 생각을 먼저 해야 합니다.

또한 erase로도 해보고 + erase 없이 해보는 연습을 하시는게 중요합니다.

제 코드처럼 erase없이도 하는 연습 꼭 해주세요.

그리고... 제가 한번 다듬어 봤는데요. ㅎㅎ 해당 부분 참고해주세요(주석도 달았습니다.)

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

const int N = 50;
vector<int> v[N];
int result = 0;

//erase의 단순화
void erase(int index) {
    for (int child : v[index]) {
        erase(child);
    }
    v[index].clear();  
}
// 리프노드 체크 단순화
void calculate(int index) {
    if (v[index].empty()) {
        result++;  
        return;
    }
    for (int child : v[index]) {
        calculate(child);
    }
}

int main() {
    int a, del, rootNum = -1;
    cin >> a;
    for (int i = 0; i < a; i++) {
        int num;
        cin >> num;
        if (num != -1) {
            v[num].push_back(i);
        } else {
            rootNum = i;
        }
    }
    cin >> del; 
    if (del == rootNum) {
        cout << 0;
        return 0;
    } 
    for (int i = 0; i < N; i++) {
        v[i].erase(remove(v[i].begin(), v[i].end(), del), v[i].end());
    } 
    erase(del);  
    calculate(rootNum);  
    cout << result;
    return 0;
}

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

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

감사합니다.

강사 큰돌 올림.