【Leetcode】1920. Build Array from Permutation

记录算法题解

于 2021-07-21 08:16:22 发布

阅读量598

点赞数

分类专栏： LC 二分、位运算与数学

本文链接：https://blog.csdn.net/qq_46105170/article/details/118957804

版权

位运算空间复杂度数组处理 LeetCode 算法优化

关键词由CSDN通过智能技术生成

LC 二分、位运算与数学专栏收录该内容

385 篇文章

订阅专栏

这篇博客介绍了一种利用位运算在常数空间复杂度内解决LeetCode问题的策略。通过保持数组元素的低10位不变，将目标值存储在高10位，然后右移10位来构建新数组。这种方法适用于元素值小于1024的情况，确保了空间效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：

https://leetcode.com/problems/build-array-from-permutation/

给定一个长 $n$ 数组 $A$ ，要求返回一个新数组 $B$ ，使得 $B [i] = A [A [i]]$ 。题目保证 $0\le A[i]< n\le 1000$ 。要求空间 $O (1)$ 。

由于 $A[i]\le 1000<1024$ ，所以其二进制表示的位数不超过 $10$ ，而int的总位数是 $32$ ，所以可以维持 $A [i]$ 的低 $10$ 位维持不动，而将 $A [A [i]]$ 存到 $A [i]$ 的 $11\sim 20$ 位那里，这样遍历完之后，每个 $A [i]$ 的低 $10$ 位存的是自己，而 $11\sim 20$ 位存的是它要变成的数。最后再将每个数右移 $10$ 位即可。代码如下：

public class Solution {
    public int[] buildArray(int[] nums) {
    	// mask二进制表示是10个1
        int mask = (1 << 10) - 1;
        for (int i = 0; i < nums.length; i++) {
            nums[i] |= (nums[nums[i]] & mask) << 10;
        }
        
        for (int i = 0; i < nums.length; i++) {
            nums[i] = nums[i] >> 10;
        }
        
        return nums;
    }
}

时空复杂度 $O (1)$ 。