加快排列[关闭]

Question

我有下一个任务。

给定整数 n (1 <= n <= 1000000) 和 k (1 <= k <= n)。

需要找到整数1, 2, 3, ..., n的任何排列p，使得排列中每两个连续整数之间的绝对差为>= k，即对于所有i的排列p需要abs(p[i] - p[i + 1]) >= k .

如果给定的n 和k 不存在这样的排列，则输出Impossible。

原始在线任务是波斯语，这就是我不提供链接的原因。

我已经实现了下一个代码。但是解决上述任务非常慢。如何提高它的速度？

from itertools import permutations

n,k=input('').split(' ')

n=int(n);k=int(k)

def check(n,k):
    n=list(n)
    N=n[:]
    result=[]
    b=n.pop(0)
    while n:
        if abs(b-n[0]) >= k:
            result.append(True)
        b=n.pop(0)
    if len(result)+1 == len(N) and all(result):
        return True
    else:
        return False

def solver(n,k):
    return (i
        for i in (permutations(range(1,n+1)))
        if check(i,k)
    )
try:
    aaa=next(solver(n,k))
    for i in aaa:
        print(i,end=' ')
except :
    print("Impossible")

Answer 1

因为最简单的优化是通过使用非常流行的NumPy库来改进check(...)函数，只需安装一次NumPy，然后通过python -m pip install numpy命令行使用下一个代码。

Try it online!

def check(n, k):
    import numpy as np
    return np.abs(np.diff(n)).min() >= k

但是，如果您的任务的解决方案应该非常非常快，那么我实施了完整的下一个新解决方案，对于大型 n（100 万甚至更多）来说应该非常快。此外，我的下一个解决方案不使用任何额外的库，例如 numpy。

主要解决函数有solve2(n, k)它返回正确答案的列表（正确性由assert仔细检查），或者答案不存在它返回None。这个求解函数的使用示例是test2() 函数，例如它为所有小的n 和k 组合创建答案。如果求解函数返回None，测试函数在打印前将其转换为字符串Impossible。

您必须按照您的案例所需的方式实现自己的test2() 变体，例如像在原始源代码中那样从用户那里获取输入n 和k，我刚刚为test2()如何使用我的代码的示例。

Try it online!

def solve2(n, k):
    if k <= 1 or n <= 1:
        p = list(range(n))
    elif n < 2 * k - 1:
        p = None
    elif 2 * k - 1 <= n <= 2 * k + 1:
        p = [None] * n
        cnt, klo = 0, -1
        if n == 2 * k + 1:
            p[0], p[1], p[2], cnt, klo = 0, k, 2 * k, 3, 0
        for i in range(k - 1, klo, -1):
            p[cnt] = i
            cnt += 1
            if cnt >= n:
                break
            p[cnt] = i + k
            cnt += 1
    else:
        if 2 * (k + 1) <= n <= 3 * (k + 1):
            kst = k + 1
        else:
            kst = k
        p = [None] * n
        cnt = 0
        for i in range(kst):
            for j in range(i, n, kst):
                p[cnt] = j
                cnt += 1

    if p is not None:
        assert len(p) == n, (len(p), n)
        p = [(e + 1) for e in p]
        assert all(abs(f - s) >= k for f, s in zip(p[:-1], p[1:]))

    return p

def test2():
    for n in range(1, 16):
        for k in range(1, n + 1):
            answer = solve2(n, k)
            answer = answer if answer is not None else 'Impossible'
            print(n, k, answer, end = '  |  ', flush = True)

test2()