输入
树的层次遍历英文逗号分隔和指定的节点。节点的值必须是整型。
输出
中序遍历指定节点的下一个节点值。所有异常情况和没有下一个节点,请输出整型 -1 。
样例输入
1,2,3,4,5,6,7,8,9
4
样例输出
9
c++代码
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>
#include <map>
#include <set>
#include <cstdlib>
#include <stack>
#include <limits>
#include<limits.h>
#include <unordered_set>
#include <sstream>
using namespace std;
void help(vector<int> v, int root, vector<int> &res, int n) {
if(root > n) {
return;
}
help(v, root * 2, res, n);
res.push_back(v[root]);
help(v, root * 2 + 1, res, v.size() - 1);
}
int main() {
string s;
vector<int> v(1, 0);
getline(cin, s);
istringstream is(s);
int inter;
char ch;
while (is >> inter)
{
v.push_back(inter);
is >> ch;
}
int k;
cin>>k;
vector<int> res;
help(v, 1, res, v.size() - 1);
bool flag = false;
for(int i = 0; i < res.size(); i++) {
if(res[i] == k && i + 1 < res.size()) {
cout << res[i + 1] << endl;
flag = true;
}
}
if(!flag)
cout << -1 << endl;
return 0;
}