完全k叉树 模拟

牛牛切木棒

https://ac.nowcoder.com/acm/contest/9224/A

完全k叉树,每层节点数量是已知的,直接推过去模拟就可以了

class Solution {
   public:
    long long tree2(int k, vector<int>& a) {
        long long ans = 0;
        int n = a.size();
        int fa = 0;
        for (int i = 1; i < n;) {                      //第i个节点
            for (int j = 0; j < k && i < n; ++j, ++i)  //每遍历k个换一次父亲
                ans += (a[fa] ^ a[i]);
            ++fa;
        }
        return ans;
    }
};
算法竞赛之路 文章被收录于专栏

整理、记录算法竞赛的好题

全部评论

相关推荐

评论
6
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务