Y 【HDU - 4705】【树形DP】

题目链接

   题目不给直接翻译，只能自己读题，讲的是给你一大堆树上的点，问你有多少个点三个点无法用一条路径表示——就是没法一笔画的三个点，一笔画问题，我们求没法一笔画的点挺难，不如就求有多少能一笔画的点，然后就用C(N, 3)减去其即可。

   怎么求没法一笔画的点？我们可以做一个假设，知道有一个点X，我们DFS得到它在一条支路上有多少子节点，那么我们用B作为中转点，其中该支路上的子儿子中任取一点，再去除了这两段内的点，去找一个，是一定能达到的。

   所以，有这样的递推方程，dp[u]是u作为根节点时，它已经遍历过的子节点支路上的节点数，对于每添加一条支路，就是会有多出来的路，ans+=(N-dp[u]-1)*dp[v]，其中u为根节点，v为子节点，N-dp[u]-1是除了这条支路加上根节点的其他可达路径，然后为了避免重复运算，所以用dp[u]，这样就不会出现重复选到同一串节点的概率了。

 

完整代码：

#pragma comment(linker, "/STACK:16777216")  //{A, B, C}以B确认，则在其子儿子中任选一个，组成一对在同一根上的儿子
#include <iostream>                         //所以是在N-tmp(已知的子儿子点)-1【是非B及B目前子节点】  *   son
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef long long ll;
const int maxN=1e5+1;
ll N, dp[maxN], ans, all;
vector<int> vt[maxN];
void dfs(int u, int pre)
{
    int len=(int)vt[u].size();
    for(int i=0; i<len; i++)
    {
        int v=vt[u][i];
        if(v==pre) continue;
        dfs(v, u);
        dp[u]+=dp[v];
        ans+=(N - dp[u] -1) * dp[v];
    }
    dp[u]++;
}
int main()
{
    while(scanf("%lld", &N)!=EOF)
    {
        for(int i=1; i<=N; i++) vt[i].clear();  memset(dp, 0, sizeof(dp));  ans=0;
        for(int i=1; i<N; i++)
        {
            int e1, e2;
            scanf("%d%d", &e1, &e2);
            vt[e1].push_back(e2);
            vt[e2].push_back(e1);
        }
        all = N * (N-1) * (N-2)/6;
        dfs(1, -1);
        printf("%lld\n", all-ans);
    }
    return 0;
}