记录一道使用bitset的题，真的太好用啦

最新推荐文章于 2025-07-04 18:36:28 发布

原创最新推荐文章于 2025-07-04 18:36:28 发布 · 267 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

C++

算法

位运算

本文介绍了一种利用bitset数据结构解决寻找一个整数的下一个更大和更小数字的方法。通过位操作，实现将二进制表示中最近的一对10转换为01或01转换为10，从而找到目标数字。文章详细展示了如何遍历比特位，进行翻转操作，并调整左侧的比特位以保持数值的接近性。

面试题0504.下一个数

记录一下bitset的使用方法：

https://www.cnblogs.com/magisk/p/8809922.html

class Solution {
   public:
    vector<int> findClosedNumbers(int num) {
        bitset<32> small(num);
        bitset<32> bigger(num);

        int s = -1;
        // small, 10 转 01，1移到左侧
        for (int i = 1; i < 32; i++) {
            if (small[i] == 1 && small[i - 1] == 0) {
                small.flip(i);
                small.flip(i - 1);
                for (int left = 0, right = i - 2; left < right;) {
                    while (left < right && small[left] == 0) left++;
                    while (left < right && small[right] == 1) right--;
                    small.flip(left);
                    small.flip(right);
                }
                s = (int)small.to_ulong();
                break;
            }
        }

        // bigger, 01转10，1移到最右侧
        int b = -1;
        for (int i = 1; i < 32; i++) {
            if (bigger[i] == 0 && bigger[i - 1] == 1) {
                bigger.flip(i);
                bigger.flip(i - 1);

                for (int left = 0, right = i - 2; left < right;) {
                    while (left < right && bigger[left] == 1) left++;
                    while (left < right && bigger[right] == 0) right--;
                    bigger.flip(left);
                    bigger.flip(right);
                }
                b = (int)bigger.to_ulong();
                break;
            }
        }

        return {b, s};
    }
};