Skip to content

Commit

Permalink
New Problem "Serialize and Deserialize Binary Tree"
Browse files Browse the repository at this point in the history
  • Loading branch information
haoel committed Nov 10, 2015
1 parent d648361 commit 065f8e8
Show file tree
Hide file tree
Showing 2 changed files with 196 additions and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ LeetCode
|---| ----- | -------- | ---------- |
|300|[Longest Increasing Subsequence](https://leetcode.com/problems/longest-increasing-subsequence/) | [C++](./algorithms/cpp/longestIncreasingSubsequence/longestIncreasingSubsequence.cpp)|Medium|
|299|[Bulls and Cows](https://leetcode.com/problems/bulls-and-cows/) | [C++](./algorithms/cpp/bullsAndCows/bullsAndCows.cpp)|Easy|
|297|[Serialize and Deserialize Binary Tree](https://leetcode.com/problems/serialize-and-deserialize-binary-tree/) | [C++](./algorithms/cpp/serializeAndDeserializeBinaryTree/SerializeAndDeserializeBinaryTree.cpp)|Medium|
|292|[Nim Game](https://leetcode.com/problems/nim-game/) | [C++](./algorithms/cpp/nimGame/nimGame.cpp)|Easy|
|290|[Word Pattern](https://leetcode.com/problems/word-pattern/) | [C++](./algorithms/cpp/wordPattern/WordPattern.cpp)|Easy|
|287|[Find the Duplicate Number](https://leetcode.com/problems/find-the-duplicate-number/) | [C++](./algorithms/cpp/findTheDuplicateNumber/findTheDuplicateNumber.cpp)|Hard|
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,195 @@
// Source : https://leetcode.com/problems/serialize-and-deserialize-binary-tree/
// Author : Hao Chen
// Date : 2015-11-10

/***************************************************************************************
*
* Serialization is the process of converting a data structure or object into a
* sequence of bits so that it can be stored in a file or memory buffer, or transmitted
* across a network connection link to be reconstructed later in the same or another
* computer environment.
*
* Design an algorithm to serialize and deserialize a binary tree. There is no
* restriction on how your serialization/deserialization algorithm should work. You
* just need to ensure that a binary tree can be serialized to a string and this string
* can be deserialized to the original tree structure.
*
* For example, you may serialize the following tree
*
* 1
* / \
* 2 3
* / \
* 4 5
*
* as "[1,2,3,null,null,4,5]", just the same as how LeetCode OJ serializes a binary
* tree. You do not necessarily need to follow this format, so please be creative and
* come up with different approaches yourself.
*
* Note: Do not use class member/global/static variables to store states. Your
* serialize and deserialize algorithms should be stateless.
*
* Credits:Special thanks to @Louis1992 for adding this problem and creating all test
* cases.
*
***************************************************************************************/

#include <iostream>
#include <sstream>
#include <vector>
#include <string>
using namespace std;

/**
* Definition for a binary tree node.
*/

struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};


class Codec {

public:
// Encodes a tree to a single string.
string serialize(TreeNode* root) {
return serialize02(root);
return serialize01(root);
}

// Decodes your encoded data to tree.
TreeNode* deserialize(string data) {
return deserialize02(data);
return deserialize01(data);
}


/*
* ===============================================================================
* Using pre-order to serialize and deserialize the tree.
* ===============================================================================
*/
public:
// Encodes a tree to a single string.
string serialize01(TreeNode* root) {
string result;
vector<TreeNode*> v;
serializeHelper(root, v);
for(int i=0; i<v.size(); i++){
result = result + (v[i]==NULL ? "#" : to_string(v[i]->val)) + " ";
}
return result;
}

// Decodes your encoded data to tree.
TreeNode* deserialize01(string data) {
vector<TreeNode*> v;
split(data, ' ', v);

int index = 0;
return deserializeHelper(v, index);

}

private:

void serializeHelper(TreeNode* root, vector<TreeNode*>& v) {
if (root==NULL) {
v.push_back(NULL);
}else{
v.push_back(root);
serializeHelper(root->left, v);
serializeHelper(root->right, v);
}
}


TreeNode* deserializeHelper(vector<TreeNode*> &v, int& index) {

if (index >= v.size() ) return NULL;

TreeNode* root = v[index++];
if (root) {
root->left = deserializeHelper(v, index);
root->right = deserializeHelper(v, index);
}
return root;
}


void split(const string &s, char delim, vector<TreeNode*> &elems) {
stringstream ss(s);
string item;
while (getline(ss, item, delim)) {
TreeNode* node = (item=="#" ? NULL : new TreeNode( stoi(item)) );
elems.push_back(node);
}

}


/*
* ===============================================================================
* Using sstream instead of vector to improve the performance
* ===============================================================================
*/

public:
// Encodes a tree to a single string.
string serialize02(TreeNode* root) {
ostringstream out;
serialize(root, out);
return out.str();
}

// Decodes your encoded data to tree.
TreeNode* deserialize02(string data) {
istringstream in(data);
return deserialize(in);
}

private:
void serialize(TreeNode* root, ostringstream& out) {
if (root==NULL) {
out << "# ";
return;
}
out << root->val << " ";
serialize(root->left, out);
serialize(root->right, out);
}

TreeNode* deserialize(istringstream& in) {
string val;
in >> val;
if (val == "#" || val.empty() ) return NULL;

TreeNode* node = new TreeNode(stoi(val));
node->left = deserialize(in);
node->right = deserialize(in);
return node;
}
};

// Your Codec object will be instantiated and called as such:
// Codec codec;
// codec.deserialize(codec.serialize(root));


int main(int argc, char** argv)
{
string s = "1 2 3 # # 4 5";
if (argc>1){
s = argv[1];
}
cout << s << endl;
Codec codec;
cout << codec.serialize(codec.deserialize(s)) << endl;

return 0;

}

0 comments on commit 065f8e8

Please sign in to comment.