LeetCode662 二叉树最大宽度

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
/*
算法思想：
    迭代的方法，采用层序遍历，注意这里使用了队列queue来辅助运算，queue里存的是一个pair，包括结点和其当前位置。在进入新一层的循环时，首先将首结点的位置保存出来当作最左位置，然后对于遍历到的结点，都更新右结点的位置，遍历一层的结点后来计算宽度更新结果res。
*/
//算法实现：

class Solution {
public:
    int widthOfBinaryTree(TreeNode* root) {
        if (!root) 
            return 0;
        int res = 0;
        queue<pair<TreeNode*,int>> q;   //包括结点和其当前位置的队列类型
        q.push({root, 1});
        while (!q.empty()) {
            int begin = q.front().second, end = 0, n = q.size(); //每层首结点位置，尾结点位置，结点个数
            for (int i = 0; i < n; ++i) {   //遍历一层
                TreeNode* t = q.front().first;  //当前结点
                end = q.front().second;   //当前结点位置
                q.pop();
                if (t->left)    //如果左子树存在，end*2
                    q.push({t->left, end * 2});
                if (t->right)   //如果右子树存在，end*2+1
                    q.push({t->right, end * 2 + 1});
            }
            res = max(res, end - begin + 1);   //求每层的宽度最大值
        }
        return res;
    }
};