leetcode 작은 함수, 문자열에 따라 두 갈래 트리 생성

10420 단어

"[1,2,3,null,null,4,5]"에 따라 대응하는 두 갈래 나무를 생성하여 디버깅하기 편리하다

#include 
#include 

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

TreeNode* getTreeNode(const string& str)
{
	string src(str);
	src.erase(std::remove(src.begin(), src.end(), '['));
	src.erase(std::remove(src.begin(), src.end(), ']'));

	std::istringstream stream(src);

	std::string field;
	std::vector<std::string> result;
	while (std::getline(stream, field, ','))
	{
		result.push_back(field);
	}
	
	TreeNode* root = new TreeNode(atoi(result[0].c_str()));
	vector<TreeNode*> vecs;
	vecs.push_back(root);

	for (int i = 1; i < result.size(); i++)
	{
		if (strstr(result[i].c_str(), "null") == nullptr)
		{
			TreeNode* n1 = new TreeNode(atoi(result[i].c_str()));

			if (i % 2 == 1) //left
			{
				vecs[(i - 1) / 2]->left = n1;
			}
			else
			{
				vecs[(i - 1) / 2]->right = n1;
			}
			vecs.push_back(n1);
		}
		else
		{
			vecs.push_back(nullptr);
		}
	}

	return root;
}
104. [두 갈래 나무의 최대 깊이] Maximum Depth of Binary Tree
Ansible 자동화 데스크탑 사용

좋은 웹페이지 즐겨찾기

개발자 우수 사이트 수집

개발자가 알아야 할 필수 사이트 100선 추천 우리는 당신을 위해 100개의 자주 사용하는 개발자 학습 사이트를 정리했습니다
best100-homepage