算有几个简单路径对

HDU 4705

题目描述：

一棵树n~1e5,求出有多少个无序对(A,B,C),使得没有一个简单路径能够一笔画过ABC.

题解：

没有简单路径,不好弄.我们更熟悉的是简单路径.一笔带过,那么就是求出有几个简单路径的对数.简单路径3个点,一定要想到中间的那个点,枚举中间的那个点是谁,那么剩下的两个分别分布在岔路上,就是正常的横着dp.

重点：

一个转化,把非简单路径转化为简单路径,之后3个点枚举强制是中间的那个点.之后树形dp

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <ctype.h>
#include <limits.h>
#include <algorithm>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <set>
#include <bitset>
#define CLR(a) memset(a, 0, sizeof(a))
#define REP(i, a, b) for(ll i = a;i < b;i++)
#define REP_D(i, a, b) for(ll i = a;i <= b;i++)

typedef long long ll;

using namespace std;

const ll maxn = 1e5 + 100;
vector<ll> G[maxn];
ll n, dp[maxn];
ll ans;
ll o;

void dfs(ll u, ll fa)//dp[u]指的u为根的点的个数
{
    ll tmp = 0;//第一个点选取的有多少种
    ll sum = 1;
    REP(i, 0, G[u].size())
    {
        ll v = G[u][i];
        if(v!=fa)
        {
            dfs(v, u);
            sum += dp[v];
            ans += (dp[v]*tmp);//强制第二个点选在V上
            tmp += dp[v];//第一个点
        }
    }
    dp[u] = sum;
    ans += (n - sum)*(tmp);
}
void solve()
{
    ans = 0;
    dfs(1, 0);
    //printf("%d\n", ans);
    o = n*(n-1)*(n-2)/6-ans;
    printf("%I64d\n", o);
}

int main()
{
   // freopen("5Ein.txt", "r", stdin);
    //freopen("5Eout.txt", "w", stdout);
    while(scanf("%I64d", &n) != EOF)
    {
        REP_D(i, 1, n)
        {
            G[i].clear();
        }
        REP_D(i, 1, n - 1)
        {
            ll a, b;
            scanf("%I64d%I64d", &a, &b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        solve();
    }
    return 0;
}