题目大意:
就是现在一个图, 其中n个点, n <= 20000, m条边, m <= 100000
其中一些边是石子路一些事乡村小路, 图保证石子路形成一棵树, 使得所有城市通过石子路能相同, 另外是一些乡村小路
现在要从图中删除一条石子路和一条乡村小路使得图变得不连通, 问有多少种删法
大致思路:
很明显是个树形DP, 对于每个石子路, 一旦被删除, 树就被切割成了一颗子树和该子树之外两个部分, 那么对应的删除乡村小路的方案就有3种情况:
1.这棵子树不和其他部分通过乡村小路相连, 那么删掉任意一条乡村小路即可
2. 这棵子树和其他部分通过1条小路相连, 那么对应删除的乡村小路只能是这条小路
3. 这棵子树和其他部分有2条或以上数量的小路相连, 那么这个状态下是没有可行方案的
于是树形DP维护一下当前子树下有多少条小路连接到该子树之外即可
要维护这个只需要对乡村小路两个端点求LCA即可, 判断他们的LCA是不是被当前子树包含住
由于每条乡村小路最多只会被询问两次LCA, 总体复杂度并不高
代码如下:
Result : Accepted Memory : 6796 KB Time : 154 ms
#include <map>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define maxn 20010
vector<int> G[maxn];
vector<int> H[maxn];
int n, m;
int st[2*maxn][20];
int pos[maxn];
int sum;
int dfn[maxn];
int fa[maxn];
void add(int x)
{
st[++sum][0] = x;
pos[x] = sum;
return;
}
void dfs(int now, int father)
{
int nex;
for(int i = 0, sz = G[now].size(); i < sz; i++) if((nex = G[now][i]) != father)
{
fa[nex] = now;
add(now);
dfn[nex] = dfn[now] + 1;
dfs(nex, now);
}
add(now);
return;
}
int Min(int x, int y)
{
return dfn[x] < dfn[y] ? x : y;
}
void initLCA()
{
for(int j = 1; (1 << j) <= sum; j++)
for(int i = 1; i + (1 << j) - 1 <= sum; i++)
st[i][j] = Min(st[i][j - 1], st[i + (1 << (j - 1))][j - 1]);
return;
}
int lca(int u, int v)
{
u = pos[u], v = pos[v];
if(u > v) swap(u, v);
int k = 0;
while((1 << (k + 1)) <= v - u + 1) k++;
return Min(st[u][k], st[v - (1 << k) + 1][k]);
}
int dp[maxn];
int gao(int now, int father)
{
int nex;
int ret = 0;
for(int i = 0, sz = G[now].size(); i < sz; i++) if((nex = G[now][i]) != father)
ret += gao(nex, now);
for(int j = 0, sz = H[now].size(); j < sz; j++)
{
int z = lca(now, H[now][j]);
if(z == now) ret--;
else ret++;
}
return dp[now] = ret;
}
int main(){
freopen( "diversion.in" , "r" , stdin );
freopen( "diversion.out" , "w" , stdout );
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++)
G[i].clear(), H[i].clear();
int u, v, t;
for(int i = 1; i <= m; i++)
{
scanf("%d %d %d", &u, &v, &t);
if(t == 1) G[u].push_back(v), G[v].push_back(u);
else H[u].push_back(v), H[v].push_back(u);
}
fa[1] = -1;
dfn[1] = sum = 0;
dfs(1, -1);
initLCA();
gao(1, -1);
int ans = 0;
for(int i = 2; i <= n; i++)
if(dp[i] == 0)
ans += m;
else if(dp[i] == 1)
ans += 1;
else ans += 0;
printf("%d\n", ans);
return 0;
}