题解 | #格雷码#

附加题Ⅰ

https://ac.nowcoder.com/acm/contest/17942/I

思路:
预处理出所有2的次幂的答案
显然有如下递推式:

void init() {
    sz[0] = 1;
    for (int i = 1; i <= 32; i++) {
        sz[i] = 2 * sz[i - 1] + 1;
    }
}

其实sz[i]的本质含义是在0-i-1位都是0的情况,翻转第i位(可能是0变1,也可能是1变0)的最少操作次数

考虑输入的数其二进制表示中“1”的奇偶性。
不妨来看2个例子(下列数为二进制表示):
100100, 1000100100
我们发现,前一个例子中,在翻转100000的过程中,会出现100100这一步(即相当于翻转了第2位)(感兴趣的朋友可以认真推推我们是怎么预处理2次幂答案,怎么有那个递推式的)
所以100100的答案为 翻转第5位的答案 - 翻转第2位的答案
而第二个例子:
1000100100
在翻转1000000000的过程中,会出现1000100000这一步,而1000100000这一步则相当于1000100100翻转了第2位。
所以答案为 翻转第2位的答案 + 翻转9位的答案 - 翻转第5位的答案。
一般的,我们记录输入数二进制表示中1的个数。
若为奇数,则在二进制表示中,从右往左枚举出现1的位数,依次为+答案 -答案 +答案......
反之则为-答案 +答案 -答案.......
代码如下:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll sz[32]; 
void init() {
    sz[0] = 1;
    for (int i = 1; i <= 32; i++) {
        sz[i] = 2 * sz[i - 1] + 1;
    }
}
ll n;
ll cnt;
ll tmp;
ll arr[32];
ll ans;
int main() {
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    init();
    cin >> n;
    while (n) {
        if (n & 1) {
            arr[++cnt] = tmp;
        }    
        tmp ++;
        n >>= 1;
    }
    if (cnt & 1) {
        for (int i = 1; i <= cnt; i += 2) {
            ans += sz[arr[i]];
        }
        for (int i = 2; i <= cnt; i += 2) {
            ans -= sz[arr[i]];
        }
    }
    else {
        for (int i = 2; i <= cnt; i += 2) {
            ans += sz[arr[i]];
        }
        for (int i = 1; i <= cnt; i += 2) {
            ans -= sz[arr[i]];
        }        
    }
    cout << ans << "\n";
    return 0;
}
全部评论

相关推荐

点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务