题解 | #牛群的二叉树排序#

这道题的主要思路：
步骤1、计算vector容器中黑牛和白牛的数量。
步骤2、根据黑牛数量创建左子树。
步骤3、根据白牛数量创建右子树。
/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <queue>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param cows int整型vector 
     * @return TreeNode类
     */
	 //创建左右子树
    TreeNode* leftnode(int value, int number){
        queue<TreeNode*> que;
        TreeNode* root = new TreeNode(number);
        TreeNode* t;
        que.push(root);
        while (--value > 0) {
		//这里先创建左节点，当value还有值时，再创建右节点
            t = new TreeNode(number);
            que.push(t);
            que.front()->left = t;
            --value;
            if (value > 0) {
                t = new TreeNode(number);
                que.push(t);
                que.front()->right = t;
            }
            que.pop();
        }
        return root;
    }
    TreeNode* sortCowsTree(vector<int>& cows) {
        // write code here
        int left = 0;
        int right = 0;
		//计算左右子树的个数；
        for (auto& x : cows) {
            if (x == 0) {
                ++left;
            }
            else {
                ++right;
            }
        }
        TreeNode* root = new TreeNode(-1);
		//左边不为0，则创建左子树
        if (left > 0) {
            root->left = leftnode(left, 0);

        }
		//右边不为0，创建右子树
        if (right > 0) {
            root->right = leftnode(right, 1); 
        }
        return root;
    }
};