Tree UVALive - 8212

題意：

有n個點，k個顏色，每個點都要被染色，相同顏色之間的邊算是被該顏色覆蓋，問有多少邊被所有顏色覆蓋

題解：

題目給的是無根樹，我們可以將1默認為根然后求所有點的子樹大小，
我們可以這么考慮，對于圖中的紅邊，紅邊兩側如果點的數(shù)量大于等于K的數(shù)量，也就是說所有顏色均可以在紅邊兩側，說明紅邊可以被所有顏色覆蓋，貢獻為1。
如何知道紅邊兩側數(shù)量？就是求子樹大小，一側為子樹的大小，另一側為n-子樹大小

代碼：

#include <bits/stdc++.h> using namespace std; #define ll long long #define INF 0x3f3f3f3f const int maxn=2e5+9; vector<int>vec[maxn]; int siz[maxn]; int getsiz(int u,int fa){for(int i=0;i<vec[u].size();i++){int v=vec[u][i];if(v==fa)continue;siz[v]=1;siz[u]+=getsiz(v,u);}return siz[u]; } int main(){int t;scanf("%d",&t);while(t--){memset(siz,0,sizeof(siz));int n,k;scanf("%d%d",&n,&k);for(int i=1;i<n;i++){int a,b;scanf("%d%d",&a,&b);vec[a].push_back(b);vec[b].push_back(a);}siz[1]=getsiz(1,0)+1;int tot=0;for(int i=1;i<=n;i++){int x,y;x=siz[i];y=n-x;if(x>=k&&y>=k)tot++;}printf("%d\n",tot);for(int i=1;i<=n;i++){vec[i].clear();}}return 0; }

總結

以上是生活随笔為你收集整理的Tree UVALive - 8212的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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