记录编号 402217 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 [ZJOI 2015] 幻想乡战略游戏 最终得分 100
用户昵称 GravatarFoolMike 是否通过 通过
代码语言 C++ 运行时间 6.829 s
提交时间 2017-05-05 14:28:21 内存使用 29.49 MiB
显示代码纯文本
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const int N=2e5+10;
struct edge{int f,t,l;}w[N];
int n,Q,head[N],next[N];
void add(int f,int t,int l){
	static int cnt=0;
	w[++cnt]=(edge){f,t,l};
	next[cnt]=head[f];
	head[f]=cnt;
}
int dis[N][20],dep[N];
int tree_size,root,big[N],size[N],h[N],fa[N];
bool vis[N];
void getroot(int x,int fa){
	size[x]=1;big[x]=0;
	for (int i=head[x];i;i=next[i]){
		int v=w[i].t;
		if (vis[v]||v==fa) continue;
		getroot(v,x);
		size[x]+=size[v];
		big[x]=max(big[x],size[v]);
	}
	if (size[x]>=tree_size/2&&big[x]<=tree_size/2) root=x;
}
void geth(int x,int fa){
	size[x]=1;
	dis[x][++h[x]]=dep[x];
	for (int i=head[x];i;i=next[i]){
		int v=w[i].t;
		if (vis[v]||v==fa) continue;
		dep[v]=dep[x]+w[i].l;
		geth(v,x);
		size[x]+=size[v];
	}
}
typedef pair<int,int> pr;
vector<pr> E[N];
void build(int x){
	vis[x]=1;dep[x]=0;geth(x,0);
	for (int i=head[x];i;i=next[i]){
		int v=w[i].t;
		if (vis[v]) continue;
		tree_size=size[v];
		getroot(v,0);
		fa[root]=x;
		E[x].push_back(pr(v,root));
		build(root);
	}
}
int cnt[N];ll up[N],sum[N];
//cnt[x]表示分治树x中军队总个数
//sum[x]表示分治树中x子树到根带权距离和
//up[x]表示子树x对sum[fa[x]]的贡献
void add(int p,int d){//给p的权增加d
	for (int x=p;x;x=fa[x]){
		cnt[x]+=d;
		sum[x]+=(ll)dis[p][h[x]]*d;
		up[x]+=(ll)dis[p][h[x]-1]*d;
	}
}
ll query(int p){//返回重心设在p的权值
	ll ans=sum[p];
	for (int x=p;fa[x];x=fa[x])
		ans+=sum[fa[x]]-up[x]+ll(cnt[fa[x]]-cnt[x])*dis[p][h[x]-1];
	return ans;
}
ll solve(int x){
	ll ans=query(x);
	for (int i=E[x].size()-1;i>=0;i--){
		ll val=query(E[x][i].first);
		if (val<ans) return solve(E[x][i].second);
	}
	return ans;
}

int main()
{
	freopen("zjoi15_tree.in","r",stdin);
	freopen("zjoi15_tree.out","w",stdout);
	scanf("%d%d",&n,&Q);
	for (int i=1;i<n;i++){
		int u,v,l;
		scanf("%d%d%d",&u,&v,&l);
		add(u,v,l);add(v,u,l);
	}
	tree_size=n;
	getroot(1,0);
	int Root=root;
	build(root);
	while (Q--){
		int x,e;
		scanf("%d%d",&x,&e);
		add(x,e);
		printf("%lld\n",solve(Root));
	}
	return 0;
}