Python 代码并不总是正确运行

您的代码中的主要思想是正确的：将它们尽可能移到左侧并将它们移到尽可能右侧。但也有错误：

1. 数组测试1 1 0使其保持不变。虽然还有一个更小的答案—— 0 0 2。原因是条件s[y]>0。这是多余的，而且只会造成妨碍。如果数组的最后一个元素为零，为什么我们拒绝将其放在那里？我们删除这个条件。

2. 数组测试0 1 0 1使其保持不变。虽然较小的答案是0 0 0 2. 原因是条件s[x]>0。由于当前位置为零而中断循环是错误的。零右侧可能还有仍可处理的数字。但s[x]不知何故需要处理零。让我们在循环中添加一个条件：

        while b>0 and s.count(0)<a-1:
            if s[x] == 0:
                x += 1 
            else:
                s[x]-=1 
                s[y]+=1 
                b-=1 
                if s[x]==0:
                    x+=1 
                if s[y]==0:
                    y-=1 

3. 条件if s[y]==0:永远不会满足。s[y]或者已经大于零或者刚刚收到一个额外的值。我们将其删除。

4. 这种情况if s[x]==0:曾经有用并且仍然有用。但循环体的开头已经存在类似的条件。我们删除它：

        x=0
        y=a-1
        while b>0 and s.count(0)<a-1:
            if s[x] == 0:
                x += 1 
            else:
                s[x]-=1 
                s[y]+=1 
                b-=1 

5. 该条件s.count(0)<a-1很有用，但可以用更简单的条件代替 - x < y。当x较少时，y您可以继续尝试转移单位，否则必须停止循环。另一个原因是新条件被认为更快。

6. if s.count(0)==a-1:它在程序开始时起什么作用？它优化了数组几乎全部为零时的情况。有必要吗？不，你可以不用它。程序会变得更慢（但仅在特殊情况下）并且更简单。删除：

n=int(input()) 
for i in range(n):
    a, b=map(int, input().split()) 
    s=list(map(int, input().split())) 
    x=0
    y=a-1
    while b>0 and x < y:
        if s[x] == 0:
            x += 1 
        else:
            s[x]-=1 
            s[y]+=1 
            b-=1 
    print(*s)

最新版本开始正常工作。她更容易。但这并不是停止的理由。

7. 为什么我们一次转移一个单位？您可以min(s[x], b)立即转移单位。事实证明，循环内的条件是不需要的：

    while b>0 and x < y:
        m = min(s[x], b)
        s[x] -= m
        s[y] += m
        b -= m
        x += 1

8. b>0不再需要这个条件了。它使程序更快一点，但也更复杂。如果它被删除，循环while可以替换为for ... range：

n=int(input()) 
for i in range(n):
    a, b=map(int, input().split()) 
    s=list(map(int, input().split())) 
    y=a-1
    for x in range(y):
        m = min(s[x], b)
        s[x] -= m
        s[y] += m
        b -= m
    print(*s)

9. 最好格式化文本，将变量引入任务中的术语：

t = int(input()) 
for _ in range(t):
    n, k = map(int, input().split()) 
    x = list(map(int, input().split())) 
    for i in range(n - 1):
        m = min(x[i], k)
        x[i] -= m
        x[n - 1] += m
        k -= m
    print(*x)

1. 您需要对代码进行更改，使其符合问题中描述的输入和输出格式。为此，您需要将行“n=int(input())”更改为“t=int(input())”以读取输入的数量。然后您必须对结果的输出进行更改。

2. 在 while 循环中，检查条件的顺序错误。首先，您必须检查数组中是否有零个元素，然后递减 b。

3. 当 s.count(0) 等于 a-1 时，您的代码不满足问题的条件。在这种情况下，您应该将整个数组输出为 0。

下面是更正后的代码，考虑到了所描述的更改：

t = int(input())
for _ in range(t):
    n, k = map(int, input().split())
    s = list(map(int, input().split()))

    if s.count(0) == n - 1:
        result = [0] * n
        result[n - 1] = max(s)
        print(*result)
    else:
        x = 0
        y = n - 1
        while b > 0 and s.count(0) < n - 1:
            if s[x] > 0 and s[y] > 0:
                s[x] -= 1
                s[y] += 1
                b -= 1
            if s[x] == 0:
                x += 1
            if s[y] == 0:
                y -= 1
        print(*s)