剑指offer之寻找文件副本-Go

题目描述

LCR 120. 寻找文件副本

设备中存有 n 个文件，文件 id 记于数组 documents。若文件 id 相同，则定义为该文件存在副本。请返回任一存在副本的文件 id。

正常算法实现

正常实现就是用集合来判断

func findRepeatDocument(documents []int) int {
    dic := map[int]bool{}
    for _, v := range documents {
        if dic[v] == true{
            return v
        }
        dic[v] == true
    }
}

我们将数字作为key，放入map中，如果遇到相同的数，当它想要存入map的时候，就会发现value已经是true，最后，返回这个值就行

第二版本实现

如果我们先排序，再直接两两对比，相同的就返回，也可以解决该问题

func findRepeatDocument(documents []int) int {
    sort.Ints(documents)
    for i:=0; i < len(documents) - 1; i++ {
        if documents[i] == documents[i+1]{
            return documents[i]
        }
    }
    return -1
}

一般在引入一个Go自带的排序方法的时候，我们会去看看他的具体实现，这个 sort.Ints 方法用的是快速排序

最终解法

该算法可以让时间复杂度为O1，空间复杂度为O1

func findRepeatDocument(documents []int) int {
    for i := 0; i < len(documents); i++ {
        if i == documents[i] {
            continue
        }
        // 对排序的数和指针后移时遇到的某个数相等时
        // 返回该数
        if  documents[i] == documents[documents[i]] {
            return documents[i]
        }
        // 直接把数字放在正确的位置上
        // 然后除非遇到已经在正确位置的值
        //否则，让i一直在0的位置，直到 documents[0] == 0
        documents[documents[i]], documents[i] = documents[i], documents[documents[i]]
        i--
    }
}