常用的位操作.md 14.0 KB
Newer Older
L
labuladong 已提交
1 2
# 常用的位运算技巧

3 4
<p align='center'>
<a href="https://github.com/labuladong/fucking-algorithm" target="view_window"><img alt="GitHub" src="https://img.shields.io/github/stars/labuladong/fucking-algorithm?label=Stars&style=flat-square&logo=GitHub"></a>
L
labuladong 已提交
5
<a href="https://appktavsiei5995.pc.xiaoe-tech.com/index" target="_blank"><img class="my_header_icon" src="https://img.shields.io/static/v1?label=精品课程&message=查看&color=pink&style=flat"></a>
6 7 8 9
<a href="https://www.zhihu.com/people/labuladong"><img src="https://img.shields.io/badge/%E7%9F%A5%E4%B9%8E-@labuladong-000000.svg?style=flat-square&logo=Zhihu"></a>
<a href="https://space.bilibili.com/14089380"><img src="https://img.shields.io/badge/B站-@labuladong-000000.svg?style=flat-square&logo=Bilibili"></a>
</p>

L
labuladong 已提交
10
![](https://labuladong.gitee.io/pictures/souyisou1.png)
11

L
labuladong 已提交
12
**通知:[数据结构精品课](https://aep.h5.xeknow.com/s/1XJHEO) 已更新到 V2.1,[手把手刷二叉树系列课程](https://aep.xet.tech/s/3YGcq3) 上线。[第 18 期每日打卡](https://aep.xet.tech/s/2PLO1n) 开始报名。另外,建议你在我的 [网站](https://labuladong.github.io/algo/) 学习文章,体验更好。**
13 14 15



L
labuladong 已提交
16 17 18 19 20 21 22 23 24
读完本文,你不仅学会了算法套路,还可以顺便解决如下题目:

| LeetCode | 力扣 | 难度 |
| :----: | :----: | :----: |
| [136. Single Number](https://leetcode.com/problems/single-number/) | [136. 只出现一次的数字](https://leetcode.cn/problems/single-number/) | 🟢
| [191. Number of 1 Bits](https://leetcode.com/problems/number-of-1-bits/) | [191. 位1的个数](https://leetcode.cn/problems/number-of-1-bits/) | 🟢
| [231. Power of Two](https://leetcode.com/problems/power-of-two/) | [231. 2 的幂](https://leetcode.cn/problems/power-of-two/) | 🟢
| [268. Missing Number](https://leetcode.com/problems/missing-number/) | [268. 丢失的数字](https://leetcode.cn/problems/missing-number/) | 🟢
| - | [剑指 Offer 15. 二进制中1的个数](https://leetcode.cn/problems/er-jin-zhi-zhong-1de-ge-shu-lcof/) | 🟢
25 26 27

**-----------**

L
labuladong 已提交
28
位操作(Bit Manipulation)可以有很多技巧,有一个叫做 Bit Twiddling Hacks 的网站收集了几乎所有位操作的黑科技玩法,网址如下:
L
labuladong 已提交
29

L
labuladong 已提交
30
http://graphics.stanford.edu/~seander/bithacks.html
L
labuladong 已提交
31 32 33 34

但是这些技巧大部分都过于晦涩,我觉得可以作为字典查阅,没必要逐条深究。但我认为那些有趣的、有用的位运算技巧,是我们每个人需要掌握的。​

所以本文由浅入深,先展示几个有趣(但没卵用)的位运算技巧,然后再汇总一些在算法题以及工程开发中常用的位运算技巧。
L
labuladong 已提交
35 36 37

### 一、几个有趣的位操作

38
1. **利用或操作 `|` 和空格将英文字符转换为小写**
L
labuladong 已提交
39

L
labuladong 已提交
40
```java
L
labuladong 已提交
41 42 43 44
('a' | ' ') = 'a'
('A' | ' ') = 'a'
```

45
2. **利用与操作 `&` 和下划线将英文字符转换为大写**
L
labuladong 已提交
46

L
labuladong 已提交
47
```java
L
labuladong 已提交
48 49 50 51
('b' & '_') = 'B'
('B' & '_') = 'B'
```

52
3. **利用异或操作 `^` 和空格进行英文字符大小写互换**
L
labuladong 已提交
53

L
labuladong 已提交
54
```java
L
labuladong 已提交
55 56 57 58
('d' ^ ' ') = 'D'
('D' ^ ' ') = 'd'
```

L
labuladong 已提交
59
以上操作能够产生奇特效果的原因在于 ASCII 编码。ASCII 字符其实就是数字,恰巧空格和下划线对应的数字通过位运算就能改变大小写。有兴趣的读者可以查 ASCII 码表自己算算,本文就不展开讲了。
L
labuladong 已提交
60

L
labuladong 已提交
61
4. **不用临时变量交换两个数**
L
labuladong 已提交
62

L
labuladong 已提交
63
```java
L
labuladong 已提交
64 65 66 67 68 69 70
int a = 1, b = 2;
a ^= b;
b ^= a;
a ^= b;
// 现在 a = 2, b = 1
```

L
labuladong 已提交
71
5. **加一**
L
labuladong 已提交
72

L
labuladong 已提交
73
```java
L
labuladong 已提交
74 75 76 77 78
int n = 1;
n = -~n;
// 现在 n = 2
```

L
labuladong 已提交
79
6. **减一**
L
labuladong 已提交
80

L
labuladong 已提交
81
```java
L
labuladong 已提交
82 83 84 85 86
int n = 2;
n = ~-n;
// 现在 n = 1
```

L
labuladong 已提交
87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
7. **判断两个数是否异号**

```java
int x = -1, y = 2;
boolean f = ((x ^ y) < 0); // true

int x = 3, y = 2;
boolean f = ((x ^ y) < 0); // false
```

如果说前 6 个技巧的用处不大,这第 7 个技巧还是比较实用的,利用的是**补码编码**的符号位。整数编码最高位是符号位,负数的符号位是 1,非负数的符号位是 0,再借助异或的特性,可以判断出两个数字是否异号。

当然,如果不用位运算来判断是否异号,需要使用 if else 分支,还挺麻烦的。你可能想利用乘积来判断两个数是否异号,但是这种处理方式容易造成整型溢出,从而出现错误。

### `index & (arr.length - 1)` 的运用

我在 [单调栈解题套路](https://labuladong.github.io/article/fname.html?fname=单调栈) 中介绍过环形数组,其实就是利用求模(余数)的方式让数组看起来头尾相接形成一个环形,永远都走不完:

```java
int[] arr = {1,2,3,4};
int index = 0;
while (true) {
    // 在环形数组中转圈
    print(arr[index % arr.length]);
    index++;
}
// 输出:1,2,3,4,1,2,3,4,1,2,3,4...
```

但模运算 `%` 对计算机来说其实是一个比较昂贵的操作,所以我们可以用 `&` 运算来求余数:

```java
int[] arr = {1,2,3,4};
int index = 0;
while (true) {
    // 在环形数组中转圈
    print(arr[index & (arr.length - 1)]);
    index++;
}
// 输出:1,2,3,4,1,2,3,4,1,2,3,4...
```

L
labuladong 已提交
129 130
> note:注意这个技巧只适用于数组长度是 2 的幂次方的情况,比如 2、4、8、16、32 以此类推。至于如何将数组长度扩展为 2 的幂次方,这也是有比较巧妙的位运算算法的,可以参考 https://graphics.stanford.edu/~seander/bithacks.html#RoundUpPowerOf2

L
labuladong 已提交
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
简单说,`& (arr.length - 1)` 这个位运算能够替代 `% arr.length` 的模运算,性能会更好一些。

那问题来了,现在是不断地 `index++`,你做到了循环遍历。但如果不断地 `index--`,还能做到环形数组的效果吗?

答案是,如果你使用 `%` 求模的方式,那么当 `index` 小于 0 之后求模的结果也会出现负数,你需要特殊处理。但通过 `&` 与运算的方式,`index` 不会出现负数,依然可以正常工作:

```java
int[] arr = {1,2,3,4};
int index = 0;
while (true) {
    // 在环形数组中转圈
    print(arr[index & (arr.length - 1)]);
    index--;
}
// 输出:1,4,3,2,1,4,3,2,1,4,3,2,1...
```

我们自己写代码一般用不到这个技巧,但在学习一些其他代码库时可能会经常看到,这里留个印象,到时候就不会懵逼了。
L
labuladong 已提交
149

L
labuladong 已提交
150
### 二、`n & (n-1)` 的运用
L
labuladong 已提交
151

L
labuladong 已提交
152
**`n & (n-1)` 这个操作在算法中比较常见,作用是消除数字 `n` 的二进制表示中的最后一个 1**
L
labuladong 已提交
153 154 155

看个图就很容易理解了:

L
labuladong 已提交
156
![](https://labuladong.gitee.io/pictures/位操作/1.png)
157 158

其核心逻辑就是,`n - 1` 一定可以消除最后一个 1,同时把其后的 0 都变成 1,这样再和 `n` 做一次 `&` 运算,就可以仅仅把最后一个 1 变成 0 了。
L
labuladong 已提交
159

L
labuladong 已提交
160
**1、计算汉明权重(Hamming Weight)**
L
labuladong 已提交
161

L
labuladong 已提交
162
这是力扣第 191 题「位 1 的个数」:
L
labuladong 已提交
163

L
labuladong 已提交
164
![](https://labuladong.gitee.io/pictures/位操作/title.png)
L
labuladong 已提交
165

L
labuladong 已提交
166 167 168 169
就是让你返回 `n` 的二进制表示中有几个 1。因为 `n & (n - 1)` 可以消除最后一个 1,所以可以用一个循环不停地消除 1 同时计数,直到 `n` 变成 0 为止。

```java
int hammingWeight(int n) {
L
labuladong 已提交
170 171 172 173 174 175 176 177 178
    int res = 0;
    while (n != 0) {
        n = n & (n - 1);
        res++;
    }
    return res;
}
```

L
labuladong 已提交
179 180 181
**2、判断一个数是不是 2 的指数**

力扣第 231 题「2 的幂」就是这个问题。
L
labuladong 已提交
182 183 184

一个数如果是 2 的指数,那么它的二进制表示一定只含有一个 1:

L
labuladong 已提交
185
```java
L
labuladong 已提交
186 187 188 189 190
2^0 = 1 = 0b0001
2^1 = 2 = 0b0010
2^2 = 4 = 0b0100
```

L
labuladong 已提交
191
如果使用  `n & (n-1)` 的技巧就很简单了(注意运算符优先级,括号不可以省略):
L
labuladong 已提交
192

L
labuladong 已提交
193 194
```java
boolean isPowerOfTwo(int n) {
L
labuladong 已提交
195 196 197 198 199
    if (n <= 0) return false;
    return (n & (n - 1)) == 0;
}
```

L
labuladong 已提交
200
### 三、`a ^ a = 0` 的运用
201

L
labuladong 已提交
202
异或运算的性质是需要我们牢记的:
203 204 205

一个数和它本身做异或运算结果为 0,即 `a ^ a = 0`;一个数和 0 做异或运算的结果为它本身,即 `a ^ 0 = a`

L
labuladong 已提交
206 207 208 209
**1、查找只出现一次的元素**

这是力扣第 136 题「只出现一次的数字」:

L
labuladong 已提交
210
![](https://labuladong.gitee.io/pictures/位操作/title1.png)
L
labuladong 已提交
211

212 213
对于这道题目,我们只要把所有数字进行异或,成对儿的数字就会变成 0,落单的数字和 0 做异或还是它本身,所以最后异或的结果就是只出现一次的元素:

L
labuladong 已提交
214 215
```java
int singleNumber(int[] nums) {
216 217 218 219 220 221 222 223
    int res = 0;
    for (int n : nums) {
        res ^= n;
    }
    return res;
}
```

L
labuladong 已提交
224 225 226 227
**2、寻找缺失的元素**

这是力扣第 268 题「丢失的数字」:

L
labuladong 已提交
228
![](https://labuladong.gitee.io/pictures/缺失元素/title.png)
L
labuladong 已提交
229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267

给一个长度为 `n` 的数组,其索引应该在 `[0,n)`,但是现在你要装进去 `n + 1` 个元素 `[0,n]`,那么肯定有一个元素装不下嘛,请你找出这个缺失的元素。

这道题不难的,我们应该很容易想到,把这个数组排个序,然后遍历一遍,不就很容易找到缺失的那个元素了吗?

或者说,借助数据结构的特性,用一个 HashSet 把数组里出现的数字都储存下来,再遍历 `[0,n]` 之间的数字,去 HashSet 中查询,也可以很容易查出那个缺失的元素。

排序解法的时间复杂度是 O(NlogN),HashSet 的解法时间复杂度是 O(N),但是还需要 O(N) 的空间复杂度存储 HashSet。

这个问题其实还有一个特别简单的解法:等差数列求和公式。

题目的意思可以这样理解:现在有个等差数列 `0, 1, 2,..., n`,其中少了某一个数字,请你把它找出来。那这个数字不就是 `sum(0,1,..n) - sum(nums)` 嘛?

```java
int missingNumber(int[] nums) {
    int n = nums.length;
    // 虽然题目给的数据范围不大,但严谨起见,用 long 类型防止整型溢出
    // 求和公式:(首项 + 末项) * 项数 / 2
    long expect = (0 + n) * (n + 1) / 2;
    long sum = 0;
    for (int x : nums) {
        sum += x;
    }
    return (int)(expect - sum);
}
```

不过,本文的主题是位运算,我们来讲讲如何利用位运算技巧来解决这道题。

再回顾一下异或运算的性质:一个数和它本身做异或运算结果为 0,一个数和 0 做异或运算还是它本身。

而且异或运算满足交换律和结合律,也就是说:

```java
2 ^ 3 ^ 2 = 3 ^ (2 ^ 2) = 3 ^ 0 = 3
```

而这道题索就可以通过这些性质巧妙算出缺失的那个元素,比如说 `nums = [0,3,1,4]`

L
labuladong 已提交
268
![](https://labuladong.gitee.io/pictures/缺失元素/1.jpg)
L
labuladong 已提交
269 270 271

为了容易理解,我们假设先把索引补一位,然后让每个元素和自己相等的索引相对应:

L
labuladong 已提交
272
![](https://labuladong.gitee.io/pictures/缺失元素/2.jpg)
L
labuladong 已提交
273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290

这样做了之后,就可以发现除了缺失元素之外,所有的索引和元素都组成一对儿了,现在如果把这个落单的索引 2 找出来,也就找到了缺失的那个元素。

如何找这个落单的数字呢,**只要把所有的元素和索引做异或运算,成对儿的数字都会消为 0,只有这个落单的元素会剩下**,也就达到了我们的目的:

```java
int missingNumber(int[] nums) {
    int n = nums.length;
    int res = 0;
    // 先和新补的索引异或一下
    res ^= n;
    // 和其他的元素、索引做异或
    for (int i = 0; i < n; i++)
        res ^= i ^ nums[i];
    return res;
}
```

L
labuladong 已提交
291
![](https://labuladong.gitee.io/pictures/缺失元素/3.jpg)
L
labuladong 已提交
292 293

由于异或运算满足交换律和结合律,所以总是能把成对儿的数字消去,留下缺失的那个元素。
L
labuladong 已提交
294

L
labuladong 已提交
295
到这里,常见的位运算差不多都讲完了。这些技巧就是会者不难难者不会,也不需要死记硬背,只要有个印象就完全够用了。
L
labuladong 已提交
296

L
labuladong 已提交
297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325


<hr>
<details>
<summary><strong>引用本文的文章</strong></summary>

 - [丑数系列算法详解](https://labuladong.github.io/article/fname.html?fname=丑数)
 - [如何同时寻找缺失和重复的元素](https://labuladong.github.io/article/fname.html?fname=缺失和重复的元素)

</details><hr>




<hr>
<details>
<summary><strong>引用本文的题目</strong></summary>

<strong>安装 [我的 Chrome 刷题插件](https://mp.weixin.qq.com/s/X-fE9sR4BLi6T9pn7xP4pg) 点开下列题目可直接查看解题思路:</strong>

| LeetCode | 力扣 |
| :----: | :----: |
| [389. Find the Difference](https://leetcode.com/problems/find-the-difference/?show=1) | [389. 找不同](https://leetcode.cn/problems/find-the-difference/?show=1) |
| - | [剑指 Offer 15. 二进制中1的个数](https://leetcode.cn/problems/er-jin-zhi-zhong-1de-ge-shu-lcof/?show=1) |

</details>



326
**_____________**
L
labuladong 已提交
327

L
labuladong 已提交
328
**《labuladong 的算法小抄》已经出版,关注公众号查看详情;后台回复关键词「**进群**」可加入算法群;回复「**全家桶**」可下载配套 PDF 和刷题全家桶**
L
labuladong 已提交
329

L
labuladong 已提交
330
![](https://labuladong.gitee.io/pictures/souyisou2.png)
L
labuladong 已提交
331 332


J
Jody Zhou 已提交
333 334
======其他语言代码======

B
brucecat 已提交
335 336 337 338 339 340 341 342
[191.位1的个数](https://leetcode-cn.com/problems/number-of-1-bits)

[231.2的幂](https://leetcode-cn.com/problems/power-of-two/)



### python

J
Jody Zhou 已提交
343
[JodyZ0203](https://github.com/JodyZ0203)提供 191. 位1的个数 Python3 解法代码:
J
Jody Zhou 已提交
344

J
Jody Zhou 已提交
345
```Python
J
Jody Zhou 已提交
346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361
class Solution:
    def hammingWeight(self, n: int) -> int:
        
        # 先定义一个count,用来存1的出现数量
        count = 0

        # 只要二进制串不等于0之前,我们用一个循环边消除1和计1的出现数量
        while n!=0:
           
            # 用labuladong在文章中所提到的 n&(n-1) 技巧来消除最后一个1
            n = n & (n-1)

            count+=1
        
        # 当二进制串全消除完之后,返回1出现的总数量
        return count
B
brucecat 已提交
362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
```



### javascript

[191.位1的个数](https://leetcode-cn.com/problems/number-of-1-bits)

```js
let hammingWeight = function(n) {
    let res = 0;
    while (n !== 0) {
        n = n & (n - 1);
        res++;
    }
    return res;
}
```

[231.2的幂](https://leetcode-cn.com/problems/power-of-two/)

```js
/**
 * @param {number} n
 * @return {boolean}
 */
let isPowerOfTwo = function(n) {
    if (n <= 0) return false;
    return (n & (n - 1)) === 0;
}
```

[136. 只出现一次的数字](https://leetcode-cn.com/problems/single-number/)

查找只出现一次的元素

```js
/**
 * @param {number[]} nums
 * @return {number}
 */
let singleNumber = function(nums) {
    let res = 0;
    for (let n of nums) {
        res ^= n;
    }
    return res;
}
```