生活随笔
收集整理的這篇文章主要介紹了
牛客多校10 - Identical Trees(dp+二分图最小权匹配)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目鏈接:點擊查看
題目大意:給出兩個同構樹 tree1 和 tree2 ,問最少需要改變多少個結點的標號,可以使得這兩棵樹相同
題目分析:直接 dfs 維護 dp 就好了,dp[ i ][ j ] 表示 tree1 中點 i 的子樹與 tree2 中點 j 的子樹相同所需要的最小代價,如果點 i 的子樹和點 j 的子樹不同構的話,那么答案設置為無窮大,最后答案就是 dp[ rt1 ][ rt2 ] 了
二分圖權匹配我用的是KM算法,隨機數據的話時間復雜度為 n^3 ,極限數據會被卡到 n^4,不過這個題目應該沒有卡KM
無窮大的話我設置的是 1e6,因為 1e6 * 500 <= 1e9 并且 500 * 500 <= 1e6 ,既不會爆 int,且大小也剛好合適
代碼:
?
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<unordered_map>
using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=1000000;const int N=510;int n;int la[N],lb[N];//頂標bool visa[N],visb[N];//訪問標記int maze[N][N];//邊權int match[N];//右部點匹配了哪一個左部點int upd[N];bool dfs(int x)
{visa[x]=true;//訪問標記:x在交錯樹中for(int i=1;i<=n;i++){if(!visb[i]){if(la[x]+lb[i]-maze[x][i]==0)//相等子圖{visb[i]=true;//訪問標記:y在交錯樹中if(!match[i]||dfs(match[i])){match[i]=x;return true;}}elseupd[i]=min(upd[i],la[x]+lb[i]-maze[x][i]);}}return false;
} int KM()
{memset(match,0,sizeof(match));for(int i=1;i<=n;i++){la[i]=-inf;lb[i]=0;for(int j=1;j<=n;j++)la[i]=max(la[i],maze[i][j]);}for(int i=1;i<=n;i++){while(1)//直到左部點找到匹配{memset(visa,false,sizeof(visa));memset(visb,false,sizeof(visb));memset(upd,inf,sizeof(upd));if(dfs(i))break;int delta=inf;for(int j=1;j<=n;j++)if(!visb[j])delta=min(delta,upd[j]);for(int j=1;j<=n;j++)//修改頂標{if(visa[j])la[j]-=delta;if(visb[j])lb[j]+=delta;}}}int ans=0;for(int i=1;i<=n;i++)ans+=maze[match[i]][i];return ans;
}vector<int>nodea[N],nodeb[N];int dp[N][N];//dp[i][j]:第一棵樹中i的子樹和第二棵樹中j的子樹匹配的最小修改次數(如果i和j的子樹不同構,dp[i][j]=inf) void dfs(int x,int y)
{if(nodea[x].size()!=nodeb[y].size())return;if(x!=y)dp[x][y]=1;elsedp[x][y]=0;if(nodea[x].empty())return;for(auto u:nodea[x])for(auto v:nodeb[y])dfs(u,v);n=nodea[x].size();for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)maze[i][j]=-dp[nodea[x][i-1]][nodeb[y][j-1]];int temp=-KM();if(temp>=inf)dp[x][y]=inf;elsedp[x][y]+=temp;
}int main()
{
#ifndef ONLINE_JUDGE
// freopen("data.in.txt","r",stdin);
// freopen("data.out.txt","w",stdout);
#endif
// ios::sync_with_stdio(false);int n,rta,rtb;scanf("%d",&n);for(int i=1;i<=n;i++){int fa;scanf("%d",&fa);if(fa==0)rta=i;elsenodea[fa].push_back(i);}for(int i=1;i<=n;i++){int fa;scanf("%d",&fa);if(fa==0)rtb=i;elsenodeb[fa].push_back(i);}for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)dp[i][j]=inf;dfs(rta,rtb);printf("%d\n",dp[rta][rtb]);return 0;
}
?
總結
以上是生活随笔為你收集整理的牛客多校10 - Identical Trees(dp+二分图最小权匹配)的全部內容,希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯,歡迎將生活随笔推薦給好友。