淀粉质板子求调
查看原帖
淀粉质板子求调
389797
Nemonade楼主2022/6/2 20:00

TLE7、8应该是某个参数写错导致复杂度伪了,但是又找不到是哪里。

#include<bits/stdc++.h>
using namespace std;
//#define int long long
#define pfor(i,x,y) for(register int i=x;i<=y;++i)
#define mfor(i,x,y) for(register int i=x;i>=y;--i)
constexpr inline int maxx(const int &x,const int &y){return x>y?x:y;}
constexpr inline int minx(const int &x,const int &y){return x<y?x:y;}
constexpr inline int absx(const int &x){return (x>0)?(x):(~x+1);}
inline int read(){
	int x=0;bool flag=false;char ch=getchar();
	while(ch<'0'||ch>'9'){
		if(ch=='-') flag=true;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return flag?~x+1:x;
}
inline void write(int x){
	if(x<0){putchar('-');x=(~x+1);}
	if(x/10) write(x/10);
	putchar((x%10)^48);
	return;
}
const int N=1e4+5,M=1e2+5;
int n,m,x,y,z,mx[N],size[N],root,dist[N],q[N],tmp[N],tot;
int head[N],nxt[N<<1],ver[N<<1],edge[N<<1],tote;
vector<int> v;
bool vis[N],out[M],res[N<<12];
inline void add(int x,int y,int z){
	ver[++tote]=y,edge[tote]=z;
	nxt[tote]=head[x],head[x]=tote;
	return;
}
inline void getroot(int x,int fa,int num){
	mx[x]=0,size[x]=1;
	for(register int i=head[x];i;i=nxt[i]){
		y=ver[i];
		if(y==fa||vis[y]) continue;
		getroot(y,x,num);
		size[x]+=size[y],mx[x]=maxx(mx[x],size[y]);
	}
	mx[x]=maxx(mx[x],num-size[x]);
	if(mx[x]<mx[root]) root=x;
	return;
}
inline void getsize(int x,int fa){
	v.push_back(dist[x]);
	for(register int i=head[x];i;i=nxt[i]){
		y=ver[i];
		if(y==fa||vis[y]) continue;
		dist[y]=dist[x]+edge[i];
		getsize(y,x);
	}
	return;
}
inline void calc(int x){
	tot=0;
	for(register int i=head[x];i;i=nxt[i]){
		y=ver[i];
		if(vis[y]) continue;
		dist[y]=edge[i];
		v.clear();
		getsize(y,-114514);
		pfor(j,0,v.size()-1) pfor(k,1,m) if(q[k]>=v[j]) out[k]|=res[q[k]-v[j]];
		pfor(j,0,v.size()-1) tmp[++tot]=v[j],res[v[j]]=true;
	}
	pfor(i,1,tot) res[tmp[i]]=false;
	return;
}
inline void solve(int x){
	vis[x]=res[0]=true;
	calc(x);
	for(register int i=head[x];i;i=nxt[i]){
		y=ver[i];
		if(vis[y]) continue;
		mx[root=0]=INT_MAX;
		getroot(y,x,size[y]);
		solve(root);
	}
	return;
}
signed main(){
	n=read(),m=read();
	pfor(i,1,n-1){
		x=read(),y=read(),z=read();
		add(x,y,z),add(y,x,z);
	}
	pfor(i,1,m) q[i]=read();
	mx[root=0]=INT_MAX;
	getroot(1,-114514,n);
	solve(root);
	pfor(i,1,m) puts(out[i]?"AYE":"NAY");
	return 0;
}
2022/6/2 20:00
加载中...