第一次写树,我感觉我写的一点问题都没有,输入几个
例子也都是正确的。但是一直WA,WA,要疯了!QAQ
#include <iostream>
#include <vector>
#include<algorithm>
#include<set>
using namespace std;
using ll = long long;
int n, d, t, t0;
set<int>ans;
struct tre {
vector<int>br;
};
tre tree[100001];
void digui(int s,int s0) {
if (s0 == 0) { return; }
int w = tree[s].br.size() ;
for (int i = 0; i < w; i++) {
ans.insert(tree[s].br[i]);
digui(tree[s].br[i], s0 - 1);
}
}
int main() {
cin >> n >> d;
for (int i = 0; i < n - 1; i++) {
cin >> t;
cin >> t0;
tree[t].br.push_back(t0);
}
digui(1, d);
cout << ans.size();
}