【Lintcode】1236. Find All Numbers Disappeared in an Array

题目地址:

https://www.lintcode.com/problem/find-all-numbers-disappeared-in-an-array/description

给定一个长 n n n且只含 1 ∼ n 1\sim n 1∼n数字的数组 A A A,其中有两个数字出现了两次,其他数字只出现一次。要求求出 1 ∼ n 1\sim n 1∼n未出现的数字。

思路是,将所有出现的数尽可能交换到其应该在的位置上,例如 3 3 3应该在 A [ 2 ] A[2] A[2]这个地方。然后再扫描一遍,找到未出现的数即可。具体请看代码,代码如下:

import java.util.ArrayList;
import java.util.List;

public class Solution {
    /**
     * @param nums: a list of integers
     * @return: return a list of integers
     */
    public List<Integer> findDisappearedNumbers(int[] nums) {
        // write your code here
        List<Integer> res = new ArrayList<>();
    
    	// 先尽可能将数字归位
        for (int i = 0; i < nums.length; i++) {
            while (nums[nums[i] - 1] != nums[i]) {
                swap(nums, nums[i] - 1, i);
            }
        }
    
    	// 然后找出位置上放的数不对的那些位置,应该放的数就是未出现的数
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != i + 1) {
                res.add(i + 1);
            }
        }
        
        return res;
    }
    
    private void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。

上一篇:[POJ 1236] Network of Schools 题解


下一篇:PS为儿童添加逼真的大力士影子