LeetCode 338 比特位计数

LeetCode 338 比特位计数

题目描述

给你一个整数 n ，对于 0 <= i <= n 中的每个 i ，计算其二进制表示中 1 的个数 ，返回一个长度为 n + 1 的数组 ans 作为答案。

示例

• 示例 1：
  输入：n = 2
  输出：[0,1,1]
  解释：
  0 --> 0
  1 --> 1
  2 --> 10

• 示例 2：
  输入：n = 5
  输出：[0,1,1,2,1,2]
  解释：
  0 --> 0
  1 --> 1
  2 --> 10
  3 --> 11
  4 --> 100
  5 --> 101

解题思路

利用 Brian Kernighan算法 x&(x-1)，该算法可以每次清除x的最右侧的一，直到x为0

代码

class Solution {
public:

    int countones(int x) {
        int cnt = 0;
        while(x > 0) {
            x &= (x-1);
            cnt ++;
        }
        return cnt;
    }

    vector<int> countBits(int n) {
        vector<int> ans(n+1);
        for(int i = 0 ; i <= n ; i ++ ) {
            ans[i] = countones(i);
        }
        return ans;
    }
};