[Leetcode] Flatten Binary Tree to Linked List

 1 /**
 2  * Definition for binary tree
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     void helper(TreeNode *root, TreeNode *&pre) {
13         if (root == NULL) return;
14         if (pre != NULL) {
15             pre->left = NULL;
16             pre->right = root;
17         }
18         pre = root;
19         TreeNode *left = root->left;
20         TreeNode *right = root->right;
21         if (left != NULL) {
22             helper(left, pre);
23         }
24         if (right != NULL) {
25             helper(right, pre);
26         }
27     }
28     void flatten(TreeNode *root) {
29         TreeNode *pre = NULL;
30         helper(root, pre);
31     }
32 };