bzoj2314: 士兵的放置（树形DP）

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#define ll long long
#define MOD(x) ((x)>=mod?(x-mod):(x))
using namespace std;
const int maxn=500010, inf=1e9, mod=1032992941;
struct poi{int too, pre;}e[maxn<<1];
int n, x, y, tot;
int last[maxn], g[maxn][3], f[maxn][3];;
void read(int &k)
{
    int f=1; k=0; char c=getchar();
    while(c<‘0‘ || c>‘9‘) c==‘-‘&&(f=-1), c=getchar();
    while(c<=‘9‘ && c>=‘0‘) k=k*10+c-‘0‘, c=getchar();
    k*=f;   
} 
inline void add(int x, int y){e[++tot]=(poi){y, last[x]}; last[x]=tot;}
void dfs(int x, int fa)
{
    f[x][1]=maxn; f[x][2]=g[x][0]=g[x][1]=g[x][2]=1;
    for(int i=last[x], too;i;i=e[i].pre)
    if((too=e[i].too)!=fa)
    {
        dfs(too, x);
        ll tmpf=min(maxn, min(f[x][1]+min(f[too][1], f[too][2]), f[x][0]+f[too][2])), tmpg=0;
        if(f[x][1]+f[too][1]==tmpf) tmpg+=g[too][1];
        if(f[x][1]+f[too][2]==tmpf) tmpg+=g[too][2], tmpg=MOD(tmpg);
        f[x][1]=tmpf; g[x][1]=1ll*g[x][1]*tmpg%mod;
        if(f[x][0]+f[too][2]==tmpf) g[x][1]+=1ll*g[x][0]*g[too][2]%mod, g[x][1]=MOD(g[x][1]);
        f[x][0]+=f[too][1]; f[x][0]=min(maxn, f[x][0]); g[x][0]=1ll*g[x][0]*g[too][1]%mod;
        tmpf=min(f[too][0], min(f[too][1], f[too][2])); tmpg=0;
        if(f[too][0]==tmpf) tmpg+=g[too][0];
        if(f[too][1]==tmpf) tmpg+=g[too][1], tmpg=MOD(tmpg);
        if(f[too][2]==tmpf) tmpg+=g[too][2], tmpg=MOD(tmpg);
        f[x][2]+=tmpf; f[x][2]=min(maxn, f[x][2]); g[x][2]=1ll*g[x][2]*tmpg%mod;
    }
}
int main()
{
    read(n);
    for(int i=1;i<n;i++) read(x), read(y), add(x, y), add(y, x);
    dfs(1, 0); 
    if(f[1][1]<f[1][2]) printf("%d\n%d", f[1][1], g[1][1]);
    else if(f[1][1]>f[1][2]) printf("%d\n%d", f[1][2], g[1][2]);
    else printf("%d\n%d", f[1][1], MOD(g[1][1]+g[1][2]));
}