牛犇犇

• title
• solution
• code

title

戳一戳

solution

直接設(shè)$dp[i][k]$表示以$i$為根時(shí)，子樹(shù)內(nèi)，邊權(quán)為$k$時(shí)的答案
（定義寫(xiě)得好復(fù)雜，可略過(guò)）
考慮對(duì)于點(diǎn)$u$，$v$是他的一個(gè)兒子，兩點(diǎn)之間的權(quán)值為$k$
①$k=0$，那么$v$的子樹(shù)內(nèi)可以走$1/0$的邊權(quán)
$$dp[u][0]+=dp[v][1]+dp[v][0]+1$$
②$k=1$，那么在$v$子樹(shù)內(nèi)就必須要一直走邊權(quán)為$1$的邊
$$dp[u][1]+=dp[v][1]+1$$
轉(zhuǎn)移方程式中的$1$表示統(tǒng)計(jì)點(diǎn)對(duì)$(u,v)$
這就是第一個(gè)$dfs$應(yīng)該做的事，默認(rèn)以$1$為根進(jìn)行一次樹(shù)$dp$

---

接下來(lái)思考第二個(gè)$dfs$換根
考慮由根$u:2$轉(zhuǎn)移到根為$v:7$的答案
首先$7$的子樹(shù)也就是綠圈圈內(nèi)的答案是不會(huì)有改變的，繼續(xù)儲(chǔ)存在$dp[v][k],k\in [0,1]$中
但是$7$的父親$2$在換根時(shí)就會(huì)變成他的兒子，也會(huì)對(duì)其造成貢獻(xiàn)也就是黃色圈圈，但是必須要把以前$7$產(chǎn)生的貢獻(xiàn)剔除
所以此時(shí)也應(yīng)該分類討論$2?7$邊權(quán)，設(shè)$u$對(duì)$v$產(chǎn)生的貢獻(xiàn)為$w$
①$k=0$
$$w=dp[u][0]+dp[u][1]?dp[v][0]?dp[v][1]?1$$
②$k=1$
$$w=dp[u][1]?dp[v][1]?1$$
綜上：
$$dp[v][k]+=w+1$$

code

#include <cstdio> #include <vector> using namespace std; #define ll long long #define N 200005 vector < pair < int, int > > G[N]; int n; ll f[N][2];void dfs1( int u, int fa ) {for( int i = 0;i < G[u].size();i ++ ) {int v = G[u][i].first, w = G[u][i].second;if( v == fa ) continue;dfs1( v, u );if( w == 1 )f[u][w] += f[v][w] + 1;elsef[u][w] += f[v][w] + f[v][!w] + 1;} }void dfs2( int u, int fa ) {for( int i = 0;i < G[u].size();i ++ ) {int v = G[u][i].first, w = G[u][i].second;if( v == fa ) continue;if( w ) f[v][w] += f[u][w] - f[v][w];else f[v][w] += f[u][w] + f[u][!w] - f[v][!w] - f[v][w];dfs2( v, u );} }int main() {scanf( "%d", &n );for( int i = 1, u, v, w;i < n;i ++ ) {scanf( "%d %d %d", &u, &v, &w );G[u].push_back( make_pair( v, w ) );G[v].push_back( make_pair( u, w ) );}dfs1( 1, 0 );dfs2( 1, 0 );ll ans = 0;for( int i = 1;i <= n;i ++ )ans += f[i][0] + f[i][1];printf( "%lld", ans );return 0; }

明明這么簡(jiǎn)單，我竟然做了這么久！！！凸(艸皿艸 )！！！

總結(jié)

以上是生活随笔為你收集整理的CF1156D 0-1-Tree（换根DP）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。