省选校内模拟D1

tot：100+10+0=110

scp基金会已介入调查

• T1
• 有一棵树，每个节点有一些机器人，管理员在一号节点，树上有一节点k为控制室，现在管理员要从1号结点出发去控制室，同时机器人会向管理员移动，机器人和管理员速度相同，求对于每个k(k属于[1,n])，管理员能与几个机器人相遇。
• 就是对于管理员路径上的每个点x，求从x开始bfs能到达且与x最短距离小于等于deep[x]的点。
• 30分 直接bfs 复杂度O(N^2)
• 100分考虑增量法，比如一条边x->y，如果知道ans[x]，那么ans[y]=ans[x]+deepsum[deep[x]*2+1]+deepsum[deep[x]*2+2];
• deepsum[deep]是y的子树中所有深度为deep的点的权值和
• 由于是子树内，可以转化成区间问题，及求一段区间深度为deep的点权和，直接上莫队即可

Code

---

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
using namespace std;
const int N=1e6;
struct node{int y,n;}e[N];
int val[N],lin[N],fa[N],pl[N],to[N],size[N],in[N],tot=0,num[N],ans[N],deepsum[N],deep[N],len=0,n,x,y;
void read(int x,int y)
{e[++len].y=y,e[len].n=lin[x],lin[x]=len;}
struct shit{int l,r,id;}duan[N];
bool cmp(shit a,shit b){return a.l/sqrt(n)==b.l/sqrt(n)? a.r<b.r:a.l<b.l;}
void dfs(int x,int f){
	fa[x]=f;
	in[x]=++tot;
	to[tot]=x;
	size[x]=1;
	for(int i=lin[x];i;i=e[i].n){
		int y=e[i].y;
		if(y==f)continue;
		deep[y]=deep[x]+1;
		dfs(y,x);
		size[x]+=size[y];
	}
}
void dfs1(int x,int f){
	deepsum[deep[x]]+=val[x];
	for(int i=lin[x];i;i=e[i].n){
		int y=e[i].y;
		if(y==f)continue;
		
		dfs1(y,x);
		
	}
}
void dfs2(int x,int f){
	for(int i=lin[x];i;i=e[i].n){
		int y=e[i].y;
		if(y==f)continue;
		ans[y]=ans[x]+pl[y];
		dfs2(y,x);
	}
}
void add(int x){deepsum[deep[x]]+=val[x];}
void del(int x){deepsum[deep[x]]-=val[x];}
int main()
{
	freopen("escape.in","r",stdin);
	freopen("escape.out","w",stdout);
	scanf("%d",&n);
	rep(i,1,n)scanf("%d",&val[i]);
	rep(i,2,n){scanf("%d%d",&x,&y);read(x,y),read(y,x);}
	deepsum[0]=val[1],ans[1]=val[1];
	dfs(1,1); 
	rep(i,2,n){
		int l=in[i];
		int r=in[i]+size[i]-1;
		duan[i-1].l=l,duan[i-1].r=r,duan[i-1].id=i;
	}
	sort(duan+1,duan+n,cmp);
	int l=1,r=n;
	rep(i,1,n)deepsum[deep[i]]+=val[i];
	rep(i,1,n-1){
		while (l>duan[i].l) add(to[--l]);
		while (r<duan[i].r) add(to[++r]);
		while (l<duan[i].l) del(to[l++]);
		while (r>duan[i].r) del(to[r--]);
		pl[duan[i].id]=deepsum[deep[fa[duan[i].id]]*2+1]+deepsum[deep[fa[duan[i].id]]*2+2];
	}
	dfs2(1,1);
	rep(i,1,n)printf("%d\n",ans[i]);
}

• 晚上2-SAT听自闭了
• 以后还是要提高基础知识的熟练度