LeetCode之Find N Unique Integers Sum up to Zero(Kotlin)

226 阅读1分钟

问题:


方法: 数组下标偶数时存放1, -1, 2, -2...,数组下标奇数时存放1, -1, 2, -2...0即可。

class FindNUniqueIntegersSumUpToZero {
    fun sumZero(n: Int): IntArray {
        var sum = 0
        val result = IntArray(n) {
            0
        }
        for (index in result.indices) {
            if (index == result.lastIndex && sum == 0) {
                result[index] = 0
            } else if (sum != 0) {
                result[index] = -sum
                sum = 0
            } else {
                result[index] = index + 1
                sum = index + 1
            }
        }
        return result
    }
}

有问题随时沟通

具体代码实现可以参考Github