M - Kill the tree 计蒜客 - 42552(2019icpc徐州/树的重心/树形dp)
生活随笔
收集整理的這篇文章主要介紹了
M - Kill the tree 计蒜客 - 42552(2019icpc徐州/树的重心/树形dp)
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
vj地址
題目大意:找到每一顆子樹的重心
思路:
樹的重心的性質(zhì):
樹的重心如果不唯一,則至多有兩個,且這兩個重心相鄰
通過連接一條端點分別在兩個樹的邊,來將兩個樹合并成一個,那么新的重心肯定是在原來這兩個樹的重心的路徑上(兩顆樹合并重心的轉(zhuǎn)移)
(應(yīng)該不會有人不知道樹的重心的定義吧)
根據(jù)這個每次合并兩個樹的時候,找到新樹的重心,就在兩個重心的路徑上,該路徑一定會經(jīng)過根節(jié)點,所以我們轉(zhuǎn)移重心的時候最多轉(zhuǎn)移到根節(jié)點就好了。
具體細(xì)節(jié)代碼有注釋
#include<bits/stdc++.h> #define INF 0x3f3f3f3f3f3f3f3f #define inf 0x3f3f3f3f #define FILL(a,b) (memset(a,b,sizeof(a))) #define re register #define lson rt<<1 #define rson rt<<1|1 #define lowbit(a) ((a)&-(a)) #define ios std::ios::sync_with_stdio(false);std::cin.tie(0);std::cout.tie(0); #define fi first #define se secondusing namespace std; typedef long long ll; typedef unsigned long long ull; typedef pair<int,int > pii; int dx[4]= {-1,1,0,0},dy[4]= {0,0,1,-1}; const ll mod=2520; const int N=2e5+10; int n; int son[N],d[N],vis[N],p[N],dp[N]; vector <int> g[200010]; vector<int> ans[N]; //轉(zhuǎn)移,只有滿足son[x]<son[u]-son[x]的才能夠轉(zhuǎn)移,x==u時就不能夠轉(zhuǎn)移了,兩棵樹轉(zhuǎn)移到最后的時候, //深度更大的那個必然是重心,不是重心的會一直走到樹根點,如果有兩個重心那一定是重心的父節(jié)點 void up(int u,int x,int y){//轉(zhuǎn)移向上爬while(x!=u&&son[x]<son[u]-son[x]){x=p[x];}while(y!=u&&son[y]<son[u]-son[y]){y=p[y];}if(d[x]>d[y]) dp[u]=x;else dp[u]=y; } void dfs(int u,int f){//樹形dpvis[u]=1;dp[u]=u;son[u]=1;p[u]=f;d[u]=d[f]+1;for(int v:g[u]){if(v==f||vis[v]) continue;dfs(v,u);son[u]+=son[v];up(u,dp[u],dp[v]);//合并兩顆樹找到重心} } int main(){scanf("%d",&n);for(int i=1;i<=n-1;i++){int u,v;scanf("%d%d",&u,&v);g[u].push_back(v);g[v].push_back(u);}dfs(1,0);for(int i=1;i<=n;i++){if(son[dp[i]]==son[i]-son[dp[i]])//判斷父節(jié)點是不是重心cout<<min(dp[i],p[dp[i]])<<" "<<max(dp[i],p[dp[i]])<<endl;else cout<<dp[i]<<endl;}return 0; } 創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎勵來咯,堅持創(chuàng)作打卡瓜分現(xiàn)金大獎總結(jié)
以上是生活随笔為你收集整理的M - Kill the tree 计蒜客 - 42552(2019icpc徐州/树的重心/树形dp)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: C - Digital Path 计蒜客
- 下一篇: 小苏打是什么水