题目:
树形DP,别忘了子树之间的情况(拐一下距离为2)。
代码如下:
#include#include #include #include using namespace std;typedef long long ll;int const maxn=2e5+5,mod=10007;int n,hd[maxn],ct,to[maxn<<1],nxt[maxn<<1];ll w[maxn],ans,sum,mxx[maxn],s[maxn];int rd(){ int ret=0,f=1; char ch=getchar(); while(ch<'0'||ch>'9'){ if(ch=='-')f=-1; ch=getchar();} while(ch>='0'&&ch<='9')ret=(ret<<3)+(ret<<1)+ch-'0',ch=getchar(); return ret*f;}void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}void dfs(int x,int f){ ll ns=0,nm=0;//不是全局变量! for(int i=hd[x],u;i;i=nxt[i]) { if((u=to[i])==f)continue; dfs(u,x); sum=(sum+s[u]*w[x])%mod; ans=max(ans,mxx[u]*w[x]); s[x]+=w[u]; mxx[x]=max(mxx[x],w[u]); sum=(sum+ns*w[u])%mod; ns+=w[u]; ans=max(ans,nm*w[u]); nm=max(nm,w[u]); }}int main(){ n=rd(); for(int i=1,x,y;i