标签:blog gis 结构 这一 etc amp 公式 回溯 pac
#include<bits/stdc++.h>
#define clean(a,i) memset(a,i,sizeof(a))
#define ll long long
#define inl inline
#define il inl void
#define it inl int
#define ill inl ll
#define re register
#define ri re int
#define rl re ll
#define mod 10007
using namespace std;
template<class T>il read(T &x) //快读
{
int f=1;char k=getchar();x=0;
for(;k>'9'||k<'0';k=getchar()) if(k=='-') f=-1;
for(;k>='0'&&k<='9';k=getchar()) x=(x<<3)+(x<<1)+k-'0';
x*=f;
}
const int MAXN = 2e5+5;
int n,u,v,head[MAXN],num_edge,val[MAXN],mx,ans;
struct Edge{
int next,to;
Edge(){}
Edge(int next,int to):next(next),to(to){} //结构体自带生成函数
}edge[MAXN<<1]; //结构体用来存边,因为是无向边,所以大小开两倍
il add_edge(int u,int v){
edge[++num_edge]=Edge(head[u],v);head[u]=num_edge;
edge[++num_edge]=Edge(head[v],u);head[v]=num_edge;
} //加边函数
il DFS(int u,int fa,int gf){ //这道题的主体部分
if(gf) mx=max(mx,val[u]*val[gf]),ans=(ans+(val[u]*val[gf])%mod)%mod; //因为gf是从 0 传下来的,所以当gf不等于 0 的时候,说明这个点有祖父了
ri maxn=0,sum=0; // maxn 用来存这个点前面的兄弟里面最大的一只, sum 用来存前面的兄弟的权值和
for(ri i=head[u];i;i=edge[i].next){
if(edge[i].to==fa) continue;
DFS(edge[i].to,u,fa); //向下递归,然后从叶节点开始回溯
ans=(ans+(sum*val[edge[i].to])%mod)%mod;
mx=max(mx,val[edge[i].to]*maxn); // 更新答案
sum+=val[edge[i].to];sum%=mod;
maxn=max(maxn,val[edge[i].to]); // 更新当前节点的数据
}
}
int main()
{
// freopen(".in","r",stdin);
// freopen(".out","w",stdout);
read(n);
for(ri i=1;i<n;i++) read(u),read(v),add_edge(u,v); //读入边
for(ri i=1;i<=n;i++) read(val[i]); //读入权值
DFS(1,0,0); //solve
printf("%d %d",mx,(ans*2)%mod); //输出,因为是任意点对,所以答案要乘二
return 0;
}
sum%=mod;
愉快收工!
标签:blog gis 结构 这一 etc amp 公式 回溯 pac
原文地址:https://www.cnblogs.com/TheShadow/p/10659494.html