求助,代码莫名TLE
查看原帖
求助,代码莫名TLE
1025153
heyechen楼主2025/7/30 13:26
#include<bits/stdc++.h>
using namespace std;
int size[10001],mxsize[10001],k[101];
bool ans[101],del[10001],len[10000001];//del用来标记删除。len是桶,用来存储距离。 
vector<int>data;
queue<int>que,nodes;
struct edge{
	int end,val;
};
vector<edge>tree[10001];
void dfs(int root,int fa)//求子树大小 
{
	nodes.push(root);
	size[root]=1;
	mxsize[root]=0;
	for(int i=0;i<tree[root].size();i++)
		if(tree[root][i].end!=fa&&!del[tree[root][i].end])
		{
			dfs(tree[root][i].end,root);
			size[root]+=size[tree[root][i].end];
			mxsize[root]=max(mxsize[root],size[tree[root][i].end]);
		}
}
int getcenter(int root)//求重心 
{
	int x,pos,mx,smx;
	dfs(root,0);
	mx=-1;
	while(!nodes.empty())
	{
		x=nodes.front();
		nodes.pop();
		smx=max(size[root]-size[x],mxsize[x]);
		if(smx>mx)
		{
			mx=smx;
			pos=x;
		}
	}
	return pos;
}
void getdis(int root,int fa,int dis)//获取根结点到其他结点的距离 
{
	data.push_back(dis);
	for(int i=0;i<tree[root].size();i++)
		if(tree[root][i].end!=fa&&!del[tree[root][i].end])
			getdis(tree[root][i].end,root,dis+tree[root][i].val);
}
void divide(int root,int m)//点分治 
{
	int x;
	root=getcenter(root);
	for(int i=0;i<tree[root].size();i++)
		if(!del[tree[root][i].end])
		{
			getdis(tree[root][i].end,root,tree[root][i].val);
			for(int j=0;j<data.size();j++)
				for(int l=1;l<=m;l++)
					if(data[j]<=k[l])
						if(len[k[l]-data[j]])
							ans[l]=true;
			for(int j=0;j<data.size();j++)
				if(data[j]<=10000000)
				{
					len[data[j]]=true;
					que.push(data[j]);
				}
			data.clear();
		}
	while(!que.empty())//清空桶 
	{
		x=que.front();
		que.pop();
		len[x]=false;
	}
	del[root]=true;
	for(int i=0;i<tree[root].size();i++)
		if(!del[tree[root][i].end])
			divide(tree[root][i].end,m);
}
int main()
{
	int n,m,u,v,w;
	edge e;
	cin>>n>>m;
	for(int i=1;i<n;i++)
	{
		cin>>u>>v>>w;
		e.val=w;
		e.end=v;
		tree[u].push_back(e);
		e.end=u;
		tree[v].push_back(e);
	}
	len[0]=true;
	for(int i=1;i<=m;i++)//离线 
		cin>>k[i];
	divide(1,m);
	for(int i=1;i<=m;i++)
		if(ans[i]||k[i]==0)//对询问为0的情况进行特判 
			cout<<"AYE"<<endl;
		else
			cout<<"NAY"<<endl;
	return 0;
}

#7#8#9#10TLE了,但我没有看出来我的代码哪里有问题。时间复杂度为O(nm log n)。

2025/7/30 13:26
加载中...