【原】每日一题C++版（树的高度）

编程是很多偏计算机、人工智能领域必须掌握的一项技能，此编程能力在学习和工作中起着重要的作用。因此小白决定开辟一个新的板块“每日一题”，通过每天一道编程题目来强化和锻炼自己的编程能力（最起码不会忘记编程）

特别说明：编程题来自“牛客网”和“领扣”以及热心小伙伴的题目。由于小白有时想锻炼某一类编程方法，所以提供的代码不一定是最优解，但是本文提供的编程代码均为通过测试代码。

树的高度

题目描述

现在有一个由有序数对组成的树，树的节点都是用数字表示，现在给定这棵树上所有的父子关系，求这棵树的高度

输入描述:

输入的第一行表示节点的个数n（1 ≤ n ≤ 1000，节点的编号为0到n-1）组成，

下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号

输出描述:

输出树的高度，为一个整数

示例

输入

5

0 1

0 2

1 3

1 4

输出

3

解析

其实这里面的树，也就是链表。如果可以使用容器将同一高度的节点放在一个容器内，这个高度内的节点的子节点放在下一层绒球内。父节点放在上一层容器。由于我们不能总是先发现父节点，因此这个容器也会在首段插入数据。因此我们需要使用两端插入数据比较快的容器，因此我们选用list容器。而这个容器内的元素也应该是一个容器（为了方便我们插入同样高度的新节点）。

这样首先我们就在每一层中寻找父节点，如果找到，就将其子节点插入（存入）下一层；如果没有找到，我们就在每一层中寻找子节点，如果找到，就将其父节点插入（存入）上一层，并且将这个节点从需要插入“树”中的数据里面删除。最后只要看这个list容器的size就可以了。

代码

#include <iostream>
#include <list>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
 int index;
 while (cin >> index)
 {
   list<vector<int>> joint;
   list<vector<int>> tree;
   for (int i = 0; i < index-1; i++)
   {
     vector<int> subjoint;
     int father, son;
     cin >> father >> son;
     subjoint.push_back(father);
     subjoint.push_back(son);
     joint.push_back(subjoint);
     if (tree.empty())
     {
       vector<int> father_t;
       father_t.push_back(father);
       vector<int> son_t;
       son_t.push_back(son);
       tree.push_back(father_t);
       tree.push_back(son_t);
     }
   }
   while (!joint.empty())
   {
     for (auto i = joint.cbegin(); i != joint.cend(); )
     {
       bool add = false;
       for (auto m = tree.begin(); m != tree.end(); m++)
       {
         int father = (*i)[0];
         int son = (*i)[1];
         auto findfather = find((*m).cbegin(), (*m).cend(), father);
         auto findson = find((*m).cbegin(), (*m).cend(), son);
         auto lastone = tree.cend();
         lastone--;
         if (findfather != (*m).cend())
         {
           if (m == lastone)
           {
             vector<int> subjoint;
             subjoint.push_back(son);
             tree.push_back(subjoint);
             i = joint.erase(i);
             add = true;
             break;
           }
           if (m != lastone)
           {
             m++;
             (*m).push_back(son);
             m--;
             i = joint.erase(i);
             add = true;
             break;
           }
         }
         else if (findson != (*m).cend())
         {
           if (m == tree.begin())
           {
             vector<int> subjoint;
             subjoint.push_back(father);
             tree.push_front(subjoint);
             i = joint.erase(i);
             add = true;
             break;
           }
           if (m == tree.begin())
           {
             m++;
             (*m).push_back(father);
             m--;
             i = joint.erase(i);
             add = true;
             break;
           }
         }
       }
       if (!add)
       {
         i++;
       }
     }
   }
   cout << tree.size() << endl;
 }
 return 0;
}