poj3417 Network 树形Dp+LCA

时间:2021-12-13 16:32:08

题意:给定一棵n个节点的树,然后在给定m条边,去掉m条边中的一条和原树中的一条边,使得树至少分为两部分,问有多少种方案。

神题,一点也想不到做法,

首先要分析出加入一条边之后会形成环,形成环的话,如果去掉该边和环上面没有被其他环覆盖的边,那么便分为两部分了。

这样只需要记录每条边被环覆盖了几次即可,

用dp[u]表示u点的父边被覆盖了几次。

每次新加进来一条边(a,b) dp[a] ++ ,dp[b] ++ , dp[lca(a,b)] -= 2;

所有边处理完之后,遍历一边此树,同时转移状态 dp[u] += dp[v];

#define maxn 100005

struct node
{
int v,next;
};
node e[maxn * ];
int cnt ;
int head[maxn * ];
int num ;
int E[maxn * + ],L[maxn * + ],R[maxn * + ];
int f[maxn * + ][];
int dp[maxn];
void init()
{
cnt = ;
memset(head,-,sizeof(head));
}
void add(int u ,int v)
{
e[cnt].v = v ;
e[cnt].next = head[u];
head[u] = cnt ++ ;
e[cnt].v = u;
e[cnt].next = head[v] ;
head[v] = cnt ++ ;
return ;
}
void dfs(int u ,int fa,int dep)
{
E[++num] = u , L[num] = dep , R[u] = num;
for(int i = head[u] ;i != -; i = e[i].next)
if(e[i].v != fa)
{
int v = e[i].v;
dfs(v, u, dep + );
E[++num] = u , L[num] = dep;
}
return;
}
void initRMQ()
{
for(int i = ; i <= num ;i ++ ) f[i][] = i;
for(int j = ; (<<j) <= num ; j++ )
for(int i = ; i + (<<j) - <= num; i ++ )
if(L[f[i][j-]] < L[f[i+(<<(j-))][j-]]) //ps:注意下标
f[i][j] = f[i][j-];
else f[i][j] = f[i+(<<(j-))][j-];
}
int lca(int a ,int b)
{
a = R[a] , b = R[b];
if(a>b) swap(a,b);
int k = (log(1.0 + b - a )/log(2.0));
if(L[f[a][k]] < L[f[b-(<<k)+][k]] )
return E[f[a][k]];
else return E[f[b-(<<k)+][k]];
}
void dfs(int u,int fa)
{
for(int i = head[u] ; i != - ; i =e[i].next)
if(e[i].v != fa)
{
int v = e[i].v;
dfs(v,u);
dp[u] += dp[v];
}
return ;
}
int main()
{
int n,m;
int u,v;
while(scanf("%d%d",&n,&m)!=EOF)
{
init();
for(int i = ; i < n - ; i ++ )
{
scanf("%d%d",&u,&v);
add(u,v);
}
num = ;
memset(R,,sizeof(R));
dfs(,,);
initRMQ();
memset(dp,,sizeof(dp));
for(int i = ; i < m ; i ++ )
{
scanf("%d%d",&u,&v);
dp[u] ++ ;
dp[v] ++ ;
dp[lca(u,v)] -= ;
}
dfs(,);
int sum ;
sum = ;
for(int i = ;i <= n ; i++ ) // ps:1为根不要选。
{
if(dp[i] == )
sum ++;
else if(dp[i] == )
sum += m;
}
printf("%d\n",sum);
}
return ;
} /*
好题:
离线LCA + 树形DP 分析:加进来一条边(a,b),会形成一条环 a - lca(a,b) - b
如果去掉(a,b)和没有被其他环覆盖的边,那么就能把树分成两部分。
只要记录dp[u]表示u的负边被环覆盖了几次
对于新加进来的一条边(a,b) dp[a] ++ , dp[b] ++ , dp[lca(a,b)] -= 2;///
这样处理为完所有新加进来的边之后,在dfs一边,跑完状态转移:dp[u] += dp[v];
这里就可以把每条环上的边都依次加1,并且因为lca(a,b)的父边不再环上,且已经提前减去了2
这样转移正好可以把减掉的2消去。 注意:这里把lca离线处理成rmq,只需要O(n)的遍历和O(nlogn)的预处理即可。 ST表 并且这里以1为根,后面的状态转移也要以1为根,否则lca就白算了。 */

poj3417