SPOJ Con-Junctions(树形dp，方案计数)

题目链接：
SPOJ Con-Junctions
题意：
给一个$n$个节点和$n-1$条边的树，要用灯点亮所有的边，每条边至少要有一个端点放盏灯就点亮了这条边，求最少的放灯数量和最少放灯数量时的方案数。方案数结果模$10007$输出。
数据范围：$n\leq 100010$
分析：
最少放灯数量很好求。
用$dp[u][0]$表示点亮$u$子树的所有边时且$u$节点不放灯的最少数量，用$dp[u][1]$表示点亮$u$子树的所有边时且$u$节点放灯时的最少数量。
状态转移方程：

$$dp[u][0]=\sum{dp[v][1]},v是u的所有儿子$$
$$dp[u][1]=1+\sum{min(dp[v][0],dp[v][1])}，v是u的所有儿子$$
那么我们接着用 $cnt[u][0]$和 $cnt[u][1]$表示对应 $dp[u][0]$和 $dp[u][1]$时的方案数。
因为 $dp[u][0]$只能从 $dp[v][1]$转移，所以 $cnt[u][0]$的状态转移是：
$$cnt[u][0]=cnt[u][0]*cnt[v][1]\% \ mod$$
而 $dp[u][1]$可以从 $dp[v][0]$和 $dp[v][1]$转移，所以需要根据 $dp[v][0]$和 $dp[v][1]$的大小确定 $cnt[u][1]$是从 $cnt[v][0]$转移还是从 $cnt[v][1]$转移，或者两者均可 $(dp[v][1]=dp[v][0]时)$。状态转移方程见代码。
时间复杂度： $O(n)$

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
using namespace std;
typedef long long ll;
const int MAX_N = 100020;
const ll mod = 10007;

int T, n, total, num;
int head[MAX_N], dp[MAX_N][5];
ll cnt[MAX_N][5];

struct Edge {
    int v, next;
} edge[MAX_N * 2];

void AddEdge(int u, int v)
{
    edge[total].v = v;
    edge[total].next = head[u];
    head[u] = total++;
}

void dfs(int u, int p)
{
    dp[u][0] = 0, dp[u][1] = 1;
    cnt[u][0] = cnt[u][1] = 1;
    for (int i = head[u]; i != -1; i = edge[i].next) {
        int v = edge[i].v;
        if (v == p) continue;
        dfs(v, u);
        dp[u][0] += dp[v][1];
        cnt[u][0] = cnt[u][0] * cnt[v][1] % mod;

        dp[u][1] += min(dp[v][1], dp[v][0]);
        if (dp[v][1] < dp[v][0]) {
            cnt[u][1] = cnt[u][1] * cnt[v][1] % mod; 
        } else if (dp[v][1] > dp[v][0]) {
            cnt[u][1] = cnt[u][1] * cnt[v][0] % mod;
        } else {
            cnt[u][1] = cnt[u][1] * (cnt[v][1] + cnt[v][0]) % mod;
        }
    }
}

int main()
{
    scanf("%d", &T);
    while (T--) {
        scanf("%d", &n);
        memset(head, -1, sizeof(head));
        total = num = 0;
        for (int i = 1; i < n; ++i) {
            int u, v;
            scanf("%d%d", &u, &v);
            AddEdge(u, v);
            AddEdge(v, u);
        }
        dfs(1, 0);
        if (dp[1][0] < dp[1][1]) {
            printf("%d %lld\n", dp[1][0], cnt[1][0]);
        } else if (dp[1][0] > dp[1][1]) {
            printf("%d %lld\n", dp[1][1], cnt[1][1]);
        } else {
            printf("%d %lld\n", dp[1][0], (cnt[1][0] + cnt[1][1]) % mod);
        }
    }
    return 0;
}