树剖+差分求助
查看原帖
树剖+差分求助
793807
SamHH0912楼主2024/4/26 21:15

rt,查了一个小时了,没发现哪有问题

#include <bits/stdc++.h>
using namespace std;
#define LL long long

const int N=200007;
int cnt,head[N];
struct qwq{int b,nxt,c1,c2;}edge[N<<1];
inline void add(int a,int b,int c1,int c2){
	edge[++cnt]={b,head[a],c1,c2};
	head[a]=cnt;
}

int dep[N],f[N],sz[N],hson[N],hnum[N];
int c1[N],c2[N];
int c,h[N],s[N],nxt[N],key[N];

inline void dfs1(int x){//第一次深搜
	for(int i=head[x];i;i=edge[i].nxt){
		int y=edge[i].b;if(dep[y]) continue;
		dep[y]=dep[x]+1,f[y]=x;
		s[++c]=y,nxt[c]=h[x],h[x]=c,key[y]=c;
		c1[c]=edge[i].c1,c2[c]=edge[i].c2;
		dfs1(y);sz[x]+=sz[y]+1;
		if(sz[y]>sz[hson[x]]) hson[x]=y,hnum[x]=key[y];
	}
}

int ctr,dfn[N],tp[N],num[N];

inline void dfs2(int x,int t){//第二次深搜
	dfn[x]=++ctr,tp[x]=t;
	if(hson[x]) num[ctr+1]=hnum[x],dfs2(hson[x],t);
	for(int i=h[x];i;i=nxt[i]){
		int y=s[i];
		if(y!=hson[x]) num[ctr+1]=i,dfs2(y,y);
	}
}

int tim[N];

inline void move(int x,int y){//移动
	while(tp[x]!=tp[y]){
		if(dep[x]<dep[y]) swap(x,y);
		tim[dfn[tp[x]]]++,tim[dfn[x]+1]--;
		x=f[tp[x]];
	}
	if(x==y) return;
	if(dep[x]>dep[y]) swap(x,y);
	tim[dfn[x]+1]++,tim[dfn[y]+1]--;
}

int solve(){
	int n;
	cin>>n;
	for(int i=1;i<n;i++){
		int a,b,c1,c2;
		cin>>a>>b>>c1>>c2;
		add(a,b,c1,c2),add(b,a,c1,c2);
	}//输入
	
	sz[0]=-1,dep[1]=1;
	dfs1(1);dfs2(1,1);//建树
	
	for(int i=1;i<n;i++) move(i,i+1);//移动
	
	LL ans=0;
	for(int i=2;i<=n;i++){
		tim[i]+=tim[i-1];
		ans+=min(tim[i]*1ll*c1[num[i]],(LL)c2[num[i]]);
	}
	cout<<ans<<'\n';//统计并输出答案
	
	return 0;
}

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	
	solve();
	
	return 0;
}
2024/4/26 21:15
加载中...