您的位置:首页 > 其它

树的同构(25 分)

2017-10-28 19:34 281 查看
给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2,则我们称两棵树是“同构”的。例如图1给出的两棵树就是同构的,因为我们把其中一棵树的结点A、B、G的左右孩子互换后,就得到另外一棵树。而图2就不是同构的。



图1



图2

现给定两棵树,请你判断它们是否是同构的。


输入格式:

输入给出2棵二叉树树的信息。对于每棵树,首先在一行中给出一个非负整数N (≤),即该树的结点数(此时假设结点从0到N−1编号);随后N行,第i行对应编号第i个结点,给出该结点中存储的1个英文大写字母、其左孩子结点的编号、右孩子结点的编号。如果孩子结点为空,则在相应位置上给出“-”。给出的数据间用一个空格分隔。注意:题目保证每个结点中存储的字母是不同的。


输出格式:

如果两棵树是同构的,输出“Yes”,否则输出“No”

先找到根节点,然后根据根节点,同时递归两棵树,如果不符合调节返回就可以,还有注意就是一个事情,自己一开始没有想到,就是这棵树只有左结点,类似一条直线。

其实有一个更简单的做法,就是不断用vector的二维数组存每个结点子结点,然后利用vector的 == 重载,就可以直接判断,对应结点的子结点是否相通,就可以判断是否是同构的了

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <math.h>
#include <stack>
#include <utility>
#include <string>
#include <sstream>
#include <cstdlib>
#include <set>
#define LL long long
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1000 + 10;
int dir[4][2] = {{1,0},{0,1},{-1,0},{0,-1}};
int Root1,Root2;
struct TreeNode
{
char Data;
int Left;
int Right;
TreeNode(char a,int l,int r)
{
Data = a;
Left = l;
Right = r;
}
};
vector<TreeNode> TB1;
vector<TreeNode> TB2;
int vis1[maxn];
int vis2[maxn];
int n,m;
int ans;
void init()
{
memset(vis1,0,sizeof(vis1));
memset(vis2,0,sizeof(vis2));
}
int judge(int a,int b)
{

if(a == -1 && b == -1)
return 1;
if(a == -1 && b != -1)
return 0;
if(a != -1 && b == -1)
return 0;
if(TB1[a].Data != TB2[b].Data)
return 0;
if(TB1[a].Left == -1 && TB2[b].Left == -1)
return judge(TB1[a].Right,TB2[b].Right);
if((TB1[a].Left != -1 && TB2[b].Left != -1)&&(TB1[TB1[a].Left].Data == TB2[TB2[b].Left].Data))
return ( judge( TB1[a].Left, TB2[b].Left )&&judge( TB1[a].Right, TB2[b].Right ) );
else
return ( judge( TB1[a].Left, TB2[b].Right )&&judge( TB1[a].Right, TB2[b].Left ) );
}
int main()
{

scanf("%d",&n);
TB1.clear();
TB2.clear();
Root1 = -1;
Root2 = -1;
for(int i = 0; i < n; i++)
{
char a,b,c;
cin>>a>>b>>c;
int left,right;
if(b != '-')
{
left = b - '0';
vis1[left] = 1;
}
else
{
left = -1;
}
if(c != '-')
{
right = c - '0';
vis1[right] = 1;
}
else
{
right = -1;
}
TB1.push_back(TreeNode(a,left,right));
}
for( int i = 0; i < n; i++)
{
if(!vis1[i])
{
Root1= i;
break;
}
}
scanf("%d",&m);
if(n != m)
{
puts("No");
return 0;
}
init();
for(int i = 0; i < m; i++)
{
char a,b,c;
cin>>a>>b>>c;
int left,right;
if(b != '-')
{
left = b - '0';
vis2[left] = 1;
}
else
{
left = -1;
}
if(c != '-')
{
right = c - '0';
vis2[right] = 1;
}
else
{
right = -1;
}
TB2.push_back(TreeNode(a,left,right));
}
for( int i = 0; i < m; i++)
{
if(!vis2[i])
{
Root2= i;
break;
}
}

if(judge(Root1,Root2))
{
cout<<"Yes"<<endl;
}
else
cout<<"No"<<endl;
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: