浙大数据结构:04-树4 是否同一棵二叉搜索树

这道题依然使用了map(不知道为什么很爱用map,哈哈),缩短了一些代码量
简单先说一下思路,建立一棵标准的树,然后依次建立待测试的树,拿它与标准的树每一个结点去对比,不一样则为No.

1、条件准备

map还是老样子,键存值,数对存左右结点
#include 
#include 
using namespace std;

map> ma, mb;
这里n,l输入并判断是否结束,具体实现放在solve里 

int main()
{
  ios::sync_with_stdio(false);
  cin.tie(0), cout.tie(0);
  int n, l;
  while (1)
  {
    cin >> n;
    if (n == 0)
      break;
    cin >> l;
    solve(n, l);
  }

  return 0;
}

2、solve函数

首先清空两棵树,然后用buildtree建立参照树,
循环遍历,建立待检测的树,然后遍历标准树中的结点,看看二者是否一样,
如果待测树未找到该结点或结点左右不一样则输出No。
f的左右是判断输出yes还是no
void solve(int n, int L)
{
  ma.clear();
  mb.clear();
  buildtree(ma, n);

  for (int i = 0; i < L; i++)
  {
    int f = 0;
    mb.clear();
    buildtree(mb, n);
    for (auto i = ma.begin(); i != ma.end(); i++)
    {
      if (mb.find(i->first) == mb.end() || i->second != mb[i->first])
        f = 1;
    }
    if (f)
      cout << "No" << endl;
    else
      cout << "Yes" << endl;
  }
}

3、buildtree函数

首先输入根节点,用m[-1].first来存。
然后循环遍历,输入结点找到其位置加进去。
t是当前所在结点,值比该节点大,看右结点是否为空,不为空遍历,为空则存储。
值比该节点小,看左结点是否为空,不为空遍历,为空则存储。
只要存储了就退出。
void buildtree(map> &m, int n)
{
  cin >> m[-1].first;
  for (int i = 1; i < n; i++)
  {
    int node;
    cin >> node;
     int t = m[-1].first;
    while (t)
    {
      if (node > t)
        if (!m[t].second)
        {
          m[t].second = node;
          break;
        }
        else
          t = m[t].second;
      else if (node < t)
        if (!m[t].first)
        {
          m[t].first = node;
          break;
        }
        else
          t = m[t].first;
    }
  }
}

4、总结

这道题也并不是很难,代码量还是不多的。
完整代码如下:
  #include 
  #include 
  using namespace std;

  map> ma, mb;

void buildtree(map> &m, int n)
{
  cin >> m[-1].first;
  for (int i = 1; i < n; i++)
  {
    int node;
    cin >> node;
     int t = m[-1].first;
    while (t)
    {
      if (node > t)
        if (!m[t].second)
        {
          m[t].second = node;
          break;
        }
        else
          t = m[t].second;
      else if (node < t)
        if (!m[t].first)
        {
          m[t].first = node;
          break;
        }
        else
          t = m[t].first;
    }
  }
}

void solve(int n, int L)
{
  ma.clear();
  mb.clear();
  buildtree(ma, n);

  for (int i = 0; i < L; i++)
  {
    int f = 0;
    mb.clear();
    buildtree(mb, n);
    for (auto i = ma.begin(); i != ma.end(); i++)
    {
      if (mb.find(i->first) == mb.end() || i->second != mb[i->first])
        f = 1;
    }
    if (f)
      cout << "No" << endl;
    else
      cout << "Yes" << endl;
  }
}

int main()
{
  ios::sync_with_stdio(false);
  cin.tie(0), cout.tie(0);
  int n, l;
  while (1)
  {
    cin >> n;
    if (n == 0)
      break;
    cin >> l;
    solve(n, l);
  }

  return 0;
}

你可能感兴趣的:(数据结构浙大,数据结构,c++,算法)