9489

예전에는 왜 이렇게 많이 틀렸을까 보니...

입력이 1 2로 시작하는 경우를 생각 안해서 그런것 같다

이런 경우는 트리가 두개 생길 수 있는데 나는 그냥 하나의 트리로 놓고 문제를 풀었으니

틀릴수밖에...

조금은 불친절한 문제인것 같다, 이런 경우도 있다는걸 알려줘도 무방 할 것 같은데

뭐 이런것도 실력이니까;

#include<iostream> #include<cstdio> #include<queue> #include<vector> using namespace std; struct Node{ int root, idx; vector<int> children; }; Node data[1005]; int main() { int n, k, bfo, cur, root, tgt_idx; queue<Node> q; Node tmp; while(1) { scanf("%d %d", &n, &k); if(n==0&&k==0) break; while(!q.empty())q.pop(); bfo=-1; data[0].children.clear(); data[0].root=-1; data[0].idx=0; q.push(data[0]); q.push(data[0]); for(int i=1; i<=n; i++) { data[i].children.clear(); scanf("%d", &cur); if(cur==k) tgt_idx=i; if(cur-bfo!=1) { q.pop(); root=q.front().idx; } data[root].children.push_back(i); data[i].root=root; data[i].idx=i; bfo=cur; q.push(data[i]); } int par, grandpar; par=data[tgt_idx].root; grandpar=data[par].root; int res=0; if(par==0 || grandpar==0) { } else { for(int i=0; i<data[grandpar].children.size(); i++) { if(data[grandpar].children[i]!=par) res+=data[data[grandpar].children[i]].children.size(); } } printf("%d\n", res); } return 0; }