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

这道题依然使用了map（不知道为什么很爱用map，哈哈），缩短了一些代码量

简单先说一下思路，建立一棵标准的树，然后依次建立待测试的树，拿它与标准的树每一个结点去对比，不一样则为No.

1、条件准备

map还是老样子，键存值，数对存左右结点

#include <iostream>
#include <map>
using namespace std;

map<int, pair<int, int>> 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<int, pair<int, int>> &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 <iostream>
  #include <map>
  using namespace std;

  map<int, pair<int, int>> ma, mb;

void buildtree(map<int, pair<int, int>> &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;
}