洛谷 1351 联合权值——树形dp

题目：https://www.luogu.org/problemnew/show/P1351

对拍了一下，才发现自己漏掉了那种拐弯的情况。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e5+,mod=;
int n,hd[N],xnt,to[N<<],nxt[N<<],w[N],s[N],mx[N],ans,prn;
int rdn()
{
  int ret=,fx=; char ch=getchar();
  while(ch>''||ch<''){ if(ch=='-')fx=-; ch=getchar();}
  while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'', ch=getchar();
  return ret*fx;
}
void add(int x,int y)
{
  to[++xnt]=y; nxt[xnt]=hd[x]; hd[x]=xnt;
  to[++xnt]=x; nxt[xnt]=hd[y]; hd[y]=xnt;
}
void upd(int &x){x-=(x>=mod?mod:);}
void dfs(int cr,int fa)
{
  int ns=,nm=;
  for(int i=hd[cr],v;i;i=nxt[i])
    if((v=to[i])!=fa)
      {
    dfs(v,cr); ans+=s[v]*w[cr]%mod;
    upd(ans);
    prn=max(prn,mx[v]*w[cr]);
    s[cr]+=w[v]; mx[cr]=max(mx[cr],w[v]);
    upd(s[cr]);

    ans+=ns*w[v]%mod; upd(ans); ns+=w[v]; upd(ns);
    prn=max(prn,nm*w[v]); nm=max(nm,w[v]);
      }
}
int main()
{
  //  freopen("lg1351-data.in","r",stdin);
  //  freopen("lg1351-zj.out","w",stdout);
  n=rdn();
  for(int i=,x,y;i<n;i++)
    {
      x=rdn(); y=rdn(); add(x,y);
    }
  for(int i=;i<=n;i++) w[i]=rdn();
  dfs(,);
  ans<<=; if(ans>=mod) ans-=mod;
  printf("%d %d\n",prn,ans);
  return ;
}