LeetCode-1609 奇偶树

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/even-odd-tree

题目描述

如果一棵二叉树满足下述几个条件，则可以称为 奇偶树 ：

二叉树根节点所在层下标为 0 ，根的子节点所在层下标为 1 ，根的孙节点所在层下标为 2 ，依此类推。
偶数下标 层上的所有节点的值都是 奇 整数，从左到右按顺序 严格递增
奇数下标 层上的所有节点的值都是 偶 整数，从左到右按顺序 严格递减
给你二叉树的根节点，如果二叉树为 奇偶树 ，则返回 true ，否则返回 false 。

 

示例 1：

 

 

 

输入：root = [1,10,4,3,null,7,9,12,8,6,null,null,2]
输出：true
解释：每一层的节点值分别是：
0 层：[1]
1 层：[10,4]
2 层：[3,7,9]
3 层：[12,8,6,2]
由于 0 层和 2 层上的节点值都是奇数且严格递增，而 1 层和 3 层上的节点值都是偶数且严格递减，因此这是一棵奇偶树。

示例 2：

 

 

 

输入：root = [5,4,2,3,3,7]
输出：false
解释：每一层的节点值分别是：
0 层：[5]
1 层：[4,2]
2 层：[3,3,7]
2 层上的节点值不满足严格递增的条件，所以这不是一棵奇偶树。

 

示例 3：

 

 

 

输入：root = [5,9,1,3,5,7]
输出：false
解释：1 层上的节点值应为偶数。

示例 4：

输入：root = [1]
输出：true

示例 5：

输入：root = [11,8,6,1,3,9,11,30,20,18,16,12,10,4,2,17]
输出：true

解题思路

树的层次遍历可以解决这个问题，由于奇数和偶数的规则不是一致的，我们可以分为奇数层和偶数层来看待这个问题。

对于奇数层，记录上一个结点的数值（初始为0）依次遍历奇数层的数据，如果出现偶数或者比上一个结点小或相等的情况，可以判断不是奇偶树，同时顺便将他们的孩子填入偶数层。遍历结束后清空奇数层以便下一次使用。

对于偶数层，记录上一个结点的数值（初始为10^6+1）依次遍历偶数层的数据，如果出现奇数或者比上一个结点大或等于的情况，可以判断不是奇偶树，同时顺便将他们的孩子填入奇数层，清空偶数层以便下一次使用。

当奇数层和偶数层都没有数据的时候，计算结束，判断这个数是奇偶树。

代码展示

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isEvenOddTree(TreeNode* root) {
        vector<TreeNode*> vzTree1, vzTree2;
        vzTree1.push_back(root);
        while(!vzTree1.empty() || !vzTree2.empty())
        {
            if(!vzTree1.empty())
            {
                int last = 0;
                for(auto node:vzTree1)
                {
                    if(node->val % 2 == 0 || node->val <= last)
                        return false;
                    last = node->val;
                    if(node->left)
                    {
                        vzTree2.push_back(node->left);
                    }
                    if(node->right)
                    {
                        vzTree2.push_back(node->right);
                    }
                }
                vzTree1.clear();
            }
            if(!vzTree2.empty())
            {
                int last = pow(10,6) + 1;
                for(auto node:vzTree2)
                {
                    if(node->val % 2 == 1 || node->val >= last)
                        return false;
                    last = node->val;
                    if(node->left)
                    {
                        vzTree1.push_back(node->left);
                    }
                    if(node->right)
                    {
                        vzTree1.push_back(node->right);
                    }
                }
                vzTree2.clear();
            }
        }
        return true;
    }
};

 

运行结果