[vijos1892]树上的最大匹配（树形DP）

 1 #include <cstdio>
 2 #include <cstring>
 3 //#include <algorithm>
 4 
 5 using namespace std;
 6 
 7 typedef long long LL;
 8 
 9 const int MaxN = 1500010;
10 
11 struct Node{
12     int v;
13     Node *nxt;
14 }pool[MaxN << 1],*tail=pool,*g[MaxN];
15 
16 int n;
17 LL m;
18 LL h[MaxN][2];
19 int fa[MaxN],f[MaxN][2];
20 LL pre[MaxN],suf[MaxN];
21 
22 inline void make_edge(int u,int v){
23     tail->v=v;tail->nxt=g[u];g[u]=tail++;
24     tail->v=u;tail->nxt=g[v];g[v]=tail++;
25 }
26 
27 inline int max(int a,int b){return a>b ? a : b;}
28 void dp(){
29     static int q[MaxN],l,r;
30     memset(fa,0xff,sizeof(fa));
31     for(fa[q[l=r=0]=1]=0;l<=r;l++)
32         for(Node *p=g[q[l]];p;p=p->nxt) if(!~fa[p->v])
33             fa[q[++r]=p->v]=q[l];
34     for(int i=r;i>=0;i--){
35         int u=q[i];
36         int maxt=0xc0c0c0c0;
37         int cnt=0,j;
38         f[u][0]=0,f[u][1]=0;
39         h[u][0]=1,h[u][1]=0;
40         for(Node *p=g[u];p;p=p->nxt) if(p->v!=fa[u]){
41             LL dt=0;
42             f[u][0]+=max(f[p->v][0],f[p->v][1]);
43             f[u][1]+=max(f[p->v][0],f[p->v][1]);
44             maxt=max(maxt,f[p->v][0]+1-max(f[p->v][0],f[p->v][1]));
45 
46             if(f[p->v][0]>f[p->v][1]) dt=h[p->v][0];
47             else if(f[p->v][0]<f[p->v][1]) dt=h[p->v][1];
48             else dt=(h[p->v][0]+h[p->v][1])%m;
49             pre[++cnt]=dt;suf[cnt]=dt;
50             (h[u][0]*=dt)%=m;
51         }
52         pre[0]=suf[cnt+1]=1;
53         for(int i=1;i<=cnt;i++) (pre[i]*=pre[i-1])%=m;
54         for(int i=cnt;i;i--) (suf[i]*=suf[i+1])%=m;
55         f[u][1]+=maxt;
56         j=1;
57         for(Node *p=g[u];p;p=p->nxt) if(p->v!=fa[u]){
58             if(f[p->v][0]+1-max(f[p->v][0],f[p->v][1])==maxt)
59                 (h[u][1]+=pre[j-1]*suf[j+1]%m*h[p->v][0]%m)%=m;
60             j++;
61         }
62     }
63     if(f[1][0]==f[1][1]) printf("%d\n%lld\n",f[1][0],(h[1][0]+h[1][1])%m);
64     else if(f[1][0]>f[1][1]) printf("%d\n%lld\n",f[1][0],h[1][0]);
65     else printf("%d\n%lld\n",f[1][1],h[1][1]);
66 }
67 int main()
68 {
69     scanf("%d",&n);
70     for(int i=1;i<n;i++){
71         int u,v;scanf("%d%d",&u,&v);
72         make_edge(u,v);
73     }
74     scanf("%lld",&m);
75     dp();
76     return 0;
77 }