Codeforces856A - Set Theory

先贴题目链接: 856A-Set Theory

题目大意是给定一个数组大小为n的数组A，A中的数的范围在[1, 1e6]且各不相同，问是否存在大小为n的数组B，满足

• 数的大小也在[1, 1e6]
• B中的数也各不相同
• A[i] + B[j] 都不相同， 对于所有的1 <= i <= n, 1 <= j <= n.

思路一:

a[i] + b[j] != a[x] + b[y], 也就是表达的 a[i] - a[x] != b[y] - b[j]，其表示B中数组任意两个数形成的差都不会在A中的(n - 1) * n / 2 个差值中存在。

所以写出了如下代码:

t = int(input())
for i in range(t):
    d = {}
    n = int(input())
    a = list(map(int, input().split()))
    #a = sorted(a
    for i in range(n):
        for j in range(0, n):
            d[a[j] - a[i]] = 1

    res = [1]
    i = 2
    while i <= 1e6 and len(res) < n:
        flag = True
        for item in res:
            if i - item in d:
                i += 1
                flag = False
                break
        if not flag:
            continue
        res.append(i)
        i += 1

    if len(res) == n:
        print("YES")
        res = list(map(str, res))
        print(' '.join(res))
    else:
        print("NO")

时间复杂度: O(n * 1e6 * n^2) = O(1e12), 超时

**思路二: **

这个是从网上看到的一个答案: 随机化的思想, 思路二中将做差的思想代替为直接进行数值模拟的思路，时间复杂度为O(n * X * n) = O(1e4 * X)[X表示找到B中n个数需要随机的次数], ac代码如下:

from random import randint

visited = [-1] * (2 * 10 ** 6 + 1)
t = int(input())
for i in range(t):
    n, A = int(input()), list(map(int, input().split()))
    res = []
    while len(res) < n:
        v = randint(1, 10**6)
        if all(visited[a + v] != i for a in A):
            for a in A:
                visited[a + v] = i
            res.append(v)
    print("YES\n" + ' '.join(map(str,res)))

这里有两个细节:

• visited 要定义在for 循环外面， 否则会tle.
• while 循环中all可以使用for 循环代替, 同时随机化可以被遍历替代，详见思路三

思路三

思路三和思路二是一样的，只不过在实现时不是使用随机化， 而是从1-1e6遍历的方式， 此时时间复杂度为O(n * 1e6 * n) = O(1e10)[
为什么这样的复杂度还能ac呢， 可能是因为测试用例中并不会达到这样的复杂度，这里我也没有想明白], ac代码如下:

visited = [-1] * (2 * 10 ** 6 + 1)
t = int(input())
for i in range(t):
    n, A = int(input()), list(map(int, input().split()))
    A.sort()
    res = []
    v = 1
    while len(res) < n:
        flag = True
        for a in A:
            if visited[a + v] == i:
                flag = False
                break
        if not flag:
            v += 1
            continue
        for a in A:
            visited[a + v] = i
        res.append(v)
        v += 1
    print("YES\n" + ' '.join(map(str,res)))

思路三的变种实现

以all或者any代替其中的for循环, 代码如下(不能ac, 时间复杂度一样，应该是卡常数):

visited = [-1] * (2 * 10 ** 6 + 1)
t = int(input())
for i in range(t):
    n, A = int(input()), list(map(int, input().split()))
    res = []
    v = 1
    while len(res) < n:
        #v = randint(1, 10**6)
        if all(visited[a + v] != i for a in A):
            for a in A:
                visited[a + v] = i
            res.append(v)
        v += 1
    print("YES\n" + ' '.join(map(str,res)))

题目的思考:

• 是否一定存在B满足要求呢? 随机化方式为什么可以找到?
• 在多大的范围内，一定可以找到满足要求的B呢? 同时， 1-1e6遍历为什么不会超时?
• 看到随机化的实现方式后，被大神雷到了，代码太简洁了，太值得学习了。

更多的codeforeces的python实现请看这里, 一块交流学习~