社区讨论

为什么开O2AC,不开O2RE

P1084[NOIP 2012 提高组] 疫情控制参与者 2已保存回复 2

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
2 条
当前快照
1 份
快照标识符
@lvq3vyka
此快照首次捕获于
2024/05/03 11:19
2 年前
此快照最后确认于
2024/05/03 14:34
2 年前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
int n, m, sum[1000010][26], a[1000010], vis[1000010], pos[1000010], p[30], fa[1000010][26], len1, len2, c[1000010];
struct nd{
	int pos, x;
}b[1000010];
struct node{
	int v, w;
};
vector<node>e[1000010];
void dfs(int x){
	for(int i=1;i<=25;i++)
		sum[x][i]=sum[fa[x][i-1]][i-1]+sum[x][i-1],fa[x][i]=fa[fa[x][i-1]][i-1];
	for(auto v:e[x]){
		if(v.v==fa[x][0])continue;
		sum[v.v][0]=v.w;
		fa[v.v][0]=x;
		dfs(v.v);
	}
}
void dfs1(int x){
	int flag=0, s=0;
	for(auto v:e[x]){
		if(v.v==fa[x][0])continue;
		dfs1(v.v);
		s++;
		if(s==1)flag=vis[v.v];
		flag=flag&&vis[v.v];
	}
	vis[x]=vis[x]||flag;
}
bool cmp(int x, int y){
	return sum[x][0]<sum[y][0];
}
bool cmp1(nd x, nd y){
	return x.x<y.x;
}
bool ck(int t){
	memset(vis,0,sizeof(vis));
	len1=len2=0;
	for(int i=1;i<=m;i++){
		int x=a[i], ti=t;
		for(int j=25;j>=0;j--){
			if(sum[x][j]<=ti&&fa[x][j]>1){
				ti-=sum[x][j];
				x=fa[x][j];
			}
		}
		if(sum[x][0]<ti)b[++len1]={pos[x],ti-sum[x][0]};
		else vis[x]=1;
	}
	dfs1(1);
	for(auto v:e[1])
		if(!vis[v.v])c[++len2]=v.v;
	sort(c+1,c+1+len2,cmp);
	sort(b+1,b+1+len1,cmp1);
	int l=1;
	for(int i=1;i<=len1;i++){
		if(!vis[b[i].pos]){
			vis[b[i].pos]=1;
			continue;
		}
		while(vis[c[l]])l++;
		if(b[i].x>=sum[c[l]][0]){
			vis[c[l]]=1;
			l++;
		}
	}
	for(auto v:e[1])
		if(!vis[v.v])return false;
	return true;
}
void build(int x){
	for(auto v:e[x]){
		if(v.v==fa[x][0])continue;
		if(x==1)pos[v.v]=v.v;
		else pos[v.v]=pos[x];
		build(v.v);
	}
}
int main(){
	p[0]=1;
	for(int i=1;i<=26;i++)
		p[i]=p[i-1]*2;
	cin>>n;
	for(int i=1;i<n;i++){
		int x, y, z;
		cin>>x>>y>>z;
		e[x].push_back({y,z});
		e[y].push_back({x,z});
	}
	dfs(1);
	cin>>m;
	for(int i=1;i<=m;i++)
		cin>>a[i];
	build(1);
	int l=0, r=300000000, ans=0;
	while(l<=r){
		int mid=(l+r)/2;
		if(ck(mid))r=mid-1,ans=mid;
		else l=mid+1;
	}
	if(ans)cout<<ans;
	else cout<<-1;
    return 0;
}

回复

2 条回复,欢迎继续交流。

正在加载回复...