蒟蒻求问:这份代码为何会超出内存限制?
查看原帖
蒟蒻求问:这份代码为何会超出内存限制?
694969
hcy1117楼主2024/2/23 21:16

rt

#include<bits/stdc++.h>
using namespace std;
int n,k;
struct node{
	int to,next; 
}e[200005];
int cnt,tot,head[100005],num[100005];
void init(){
	memset(e,0,sizeof(e));
	memset(head,0,sizeof(head));
	memset(num,1,sizeof(num));
	cnt=tot=0;
}
void add(int u,int v){
	++cnt;
	e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}
void dfs(int x)
{
	for(int i=head[x];i;i=e[i].next)
	{
		dfs(e[i].to);
		num[x]+=num[e[i].to];
	}
	if(num[x]==k){
		num[x]=0;
		++tot;
	}
}
int main()
{
	int t;
	cin>>t;
	while(t--){
		init();
		cin>>n>>k; 
		if(n%k!=0||n/k!=k){
			cout<<"NO"<<endl;
			continue;
		}
		for(int i=1;i<=n-1;i++){
			int u,v;
			cin>>u>>v;
			add(u,v);
			add(v,u);
		}
		dfs(1);
		if(tot==n/k)cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
	} 
	return 0;
}










2024/2/23 21:16
加载中...