求助,求助!QAQ
查看原帖
求助,求助!QAQ
1091996
HoshinoAiQAQ楼主2023/10/18 11:14

第一次写树,我感觉我写的一点问题都没有,输入几个 例子也都是正确的。但是一直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();
}


2023/10/18 11:14
加载中...