社区讨论
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 条回复,欢迎继续交流。
正在加载回复...