原题题目
https://codeforces.com/contest/1945/problem/D
思路:
从最后一个人一直遍历到第m + 1 个人,算出换到m的最小花费,最后加上第m个a[i]就是答案了
(真的是正确答案吗?)
很明显有一种特殊情况没有讨论,我第一次就是这样成功WA掉了
比如下面这种情况,换到m就不是最优选择,而是换到m - 1上并且给m贿赂钱
1 5 2 1 3 3 6 9 1 1 3 8 5
所以我们到第m个位置的时候不能停止,应该继续往前寻找最优解
代码如下
t = int(input())
for _ in range(t):
n, m = map(int, input().split())
lst_a = list(map(int, input().split()))
lst_b = list(map(int, input().split()))
sum = 0
ans = float("inf")
for i in range(m, n):
sum += min(lst_a[i], lst_b[i]) # 计算换到位置m的最小花费
for i in range(m - 1, -1, -1):
ans = min(ans, lst_a[i] + sum)
sum += min(lst_a[i], lst_b[i])
print(ans)