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;
}