C - Sowing Stones（python解）-atcoder

合集 - Competitive Programming(10)

1.B. Alice's Adventures in Permuting （python解）-codeforces2024-11-142.C. 小红打怪 （python解）-牛客2024-11-103.B. Replacement （python解）-codeforces2024-11-10

4.C - Sowing Stones（python解）-atcoder2024-11-10

5.C. Alya and Permutation（python解）-codeforces2024-11-026.AtCoder Beginner Contest 375 C题 (python解)2024-10-237.C. Penchick and BBQ Buns （python解）-codeforces2024-11-168.D - Strange Mirroring（python解）——Atcoder2024-11-179.D. Sharky Surfing （python解）-codeforces2024-11-1910.牛客小白月赛105 (Python题解) A~E2024-11-23

收起

C - Sowing Stones** （python解）-atcoder

原题链接：

C - Sowing Stones

问题分析：

每个包含石头的单元格 X[i] 中有 A[i] 个石头。
我们需要确保每个单元格从 1 到 N 最终都有 1 个石头。

思路:

• 可用的石头总数必须等于单元格的总数。即需要 N 个石头，但只有 ΣA[i]（其中 A[i] 是单元格 X[i] 中的石头数量）个石头。如果 ΣA[i] 小于 N，则无法满足条件，输出 -1。
• 如果 ΣA[i] 大于 N，则我们有多余的石头，需要正确分配。
• 我们只能将石头从单元格 i 移动到单元格 i+1。因此，如果某个单元格有超过 1 个石头，我们可以将多余的石头向右移动，直到耗尽该单元格的石头或填满下一个单元格。
• 我们需要跟踪确保每个单元格从 1 到 N 最终都有 1 个石头所需的操作（石头移动）次数。

代码：

​

N,M = map(int, input().split())
X = list(map(int, input().split()))
A = list(map(int, input().split()))
if sum(A) != N:# 检查所有石头的总数是否等于单元格的数量 N  
    print(-1)
    exit()
num, ans = 0, N * (N + 1) // 2# 初始化 ans 为 N*(N+1)//2，表示理想情况下每个单元格都有 1 个石头的总移动成本 
for x, a in sorted(zip(X, A), key=lambda x: x[0]):
# 如果当前可用的石头数量小于目标位置 x 前的单元格数量  
    if num < x - 1:
        ans = -1
        break
    num += a
    ans -= x * a
print(ans)