感觉写的真啰嗦
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
struct TreeLinkNode
{
int val;
TreeLinkNode *left, *right, *next;
TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
};
queue<vector<TreeLinkNode*> > q;
class Solution {
public:
void connect(TreeLinkNode *root)
{
bool first = true;
if(!root)
return;
vector<TreeLinkNode*> v1;
v1.push_back(root);
q.push(v1);
TreeLinkNode *pre = NULL;
while(!q.empty())
{
vector<TreeLinkNode*> t = q.front();
q.pop();
vector<TreeLinkNode*>::iterator iter;
vector<TreeLinkNode*> vv;
first = true;
for(iter = t.begin(); iter != t.end(); ++iter)
{
if(first)
{
pre = *iter;
first = false;
if(pre->left)
vv.push_back(pre->left);
if(pre->right)
vv.push_back(pre->right);
continue;
}
TreeLinkNode *p = *iter;
pre->next = p;
p->next = NULL;
pre = p;
if(p->left)
vv.push_back(p->left);
if(p->right)
vv.push_back(p->right);
}
pre->next = NULL;
if (vv.size() > 0)
{
q.push(vv);
}
else
break;
}
}
};
int main(void)
{
Solution s;
TreeLinkNode *root = new TreeLinkNode(1);
TreeLinkNode *node1 = new TreeLinkNode(2); root->left = node1;
TreeLinkNode *node2 = new TreeLinkNode(3); root->right = node2;
TreeLinkNode *node3 = new TreeLinkNode(4); node1->left = node3;
TreeLinkNode *node4 = new TreeLinkNode(5); node1->right = node4;
TreeLinkNode *node5 = new TreeLinkNode(6); node2->left = node5;
TreeLinkNode *node6 = new TreeLinkNode(7); node2->right = node6;
s.connect(root);
cout << node4->next->val << endl;
return 0;
}