灵异输出
查看原帖
灵异输出
315205
Kniqht楼主2023/10/31 09:36

rt,样例基本对了,但是。。。

为什么13,16,15,10像是输出把十位给砍掉了??

感觉不是太大的问题,有没有人能帮我调一下/kel

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int N=1e6+10,inf=0x3f3f3f3f;
int n,m;
int h[N],e[N],ne[N],idx;
ll f[N],w[N],f1[N];
ll mx[N],mx1[N],up[N];
bool st[N];
vector<ll> g[N];
void add(int a,int b,int c){
	e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
void dfs(int u,int fa){
	if(st[u]) mx[u]=0;
	for(int i=h[u];~i;i=ne[i]){
		int j=e[i];
		if(j==fa)continue;
		dfs(j,u);
		f[u]+=f[j];
		if(mx[j]>=0){
			f[u]+=2*w[i];
			if(mx[j]+w[i]>mx[u]) mx1[u]=mx[u],mx[u]=mx[j]+w[i];
			else mx1[u]=max(mx1[u],mx[j]+w[i]);
		}
	}
	for(int i=h[u];~i;i=ne[i]){
		int j=e[i];
		if(j==fa) continue;
		ll t=f[u]-f[j];
		if(mx[j]>=0) t-=2*w[i];
		g[u].push_back(t);
	}
} 
void dp(int u,int fa){
	for(int i=h[u],nw=0;~i;i=ne[i],nw++){
		int j=e[i];
		if(j==fa) continue;
		f1[j]=g[u][nw]+2*w[i]+f[j];
		if(mx[u]==mx[j]+w[i]&&mx1[u]>0) up[j]=mx1[u]+w[i];
		else if(mx[u]>0) up[j]=mx[u]+w[i];
		dp(j,u);
	}
}
signed main(){
	memset(h,-1,sizeof(h));
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) mx[i]=mx1[i]=up[i]=-inf;
	int u,v,x;
	for(int i=1;i<n;i++){
		scanf("%d%d%d",&u,&v,&x);
		add(u,v,x);add(v,u,x);
	}
	while(m--){
		scanf("%d",&x);
		st[x]=1;
	}
	dfs(1,-1);
	f1[1]=f[1];
	dp(1,-1);
	for(int i=1;i<=n;i++)
		printf("%lld\n",f1[i]-max(up[i],mx[i]));
    return 0;   
}
2023/10/31 09:36
加载中...