社区讨论

60暴力只能拿10分,求助

P1351[NOIP 2014 提高组] 联合权值参与者 3已保存回复 2

讨论操作

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

当前回复
2 条
当前快照
1 份
快照标识符
@lo8lfuw1
此快照首次捕获于
2023/10/27 20:32
2 年前
此快照最后确认于
2023/10/27 20:32
2 年前
查看原帖
CPP
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<map>
const int mod=10007;
const int maxn=2005;

using namespace std;

int n,t;
long long sum,ans;
long long a[maxn];
int hd[maxn],f[maxn][25],dep[maxn];
bool vis[maxn][maxn];
struct edge{
	int u,v,nxt;	
}e[maxn*2];
void addedge(int u,int v){
	e[++t].u=u;
	e[t].v=v;
	e[t].nxt=hd[u];
	hd[u]=t;
}
void dfs(int u,int fa){
	f[u][0]=fa;
	for(int i=hd[u];i;i=e[i].nxt){
		int v=e[i].v;
		if(v==fa)
			continue;
		dep[v]=dep[u]+1;
		dfs(v,u);
	}
}
void init(){
	for(int j=1;j<=20;j++)
		for(int i=1;i<=n;i++)
			f[i][j]=f[f[i][j-1]][j-1];
}
int lca(int x,int y){
	
	if(x>y)swap(x,y);//x<=y;
	for(int i=20;i;i--)
		if(dep[f[y][i]]>=dep[x])
			y=f[y][i];
	if(x==y)
		return x;
	for(int i=20;i;i--)
		if(f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
	
}
int getd(int x,int y){
	return abs(dep[x]-dep[lca(x,y)])+abs(dep[y]-dep[lca(x,y)]);
}

//void work(int u,int fa){
//	long long sumnow=0;
//	for(int i=hd[fa];i;i=e[i].nxt){
//		int v=e[i].v;
////		if(v==fa)
////			continue;
//		sumnow+=a[v];			
//	}
//	for(int i=hd[fa];i;i=e[i].nxt){
//		int v=e[i].v;
////		if(v==fa)
////			continue;
//		sum=(sum%mod+((sumnow-a[v])*a[v]%mod))%mod;		
//		ans=max((sumnow-a[v])*a[v],ans);
//	}	
//	
////	sum=(sum%mod+a[f[u][1]]*a[u]%mod)%mod;
////	ans=max(ans,a[f[u][1]]*a[u]);
//}
int main(){
	scanf("%d",&n);
	for(int i=1,u,v;i<=n-1;i++){
		scanf("%d%d",&u,&v);
		addedge(u,v);
		addedge(v,u);
	}
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	dfs(1,0);
	init();
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(getd(i,j)==2){
				ans=max(ans,a[i]*a[j]);
				sum=(sum%mod+1LL*a[i]*a[j]%mod)%mod;
				//vis[i][j]=1;
			}
		}
	}
//	for(int i=1;i<=n;i++)
//		work(i,f[i][0]);
	printf("%lld %lld",ans,sum%mod);
	return 0;
}

回复

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

正在加载回复...