小b和排序(51Nod-2484)
题目
小b有两个长度都为n的序列A,B。
现在她需要选择一些i,然后交换A[i]和B[i],使得A和B都变成严格递增的序列。
你能帮小b求出最少交换次数吗?
输入保证有解。
输入
第一行输入一个正整数n,表示两个数组的长度;
第二行输入n个数,表示A[i],以空格隔开;
第三行输入n个数,表示B[i],以空格隔开;
其中1≤n≤1000, 0≤A[i],B[i]≤2000
输出
输出一个数,表示交换次数
输入样例
4
1 3 5 4
1 2 3 7
输出样例
1
如何插入一段漂亮的代码片
去博客设置页面,选择一款你喜欢的代码片高亮样式,下面展示同样高亮的 代码片
.
import math
N = int(input())
A = list(map(int, input().split()))
#保证下标从1开始
A.insert(0, -1)
B = list(map(int, input().split()))
B.insert(0, -1)
#初始化为无穷大
dp = [[math.inf] * (N+2) for _ in range(N+2)]
#保证下标从1开始
dp[0][0] = 0
dp[0][1] = 0
for i in range(1, N+1):
#只看局部,保证第i个元素>第i-1个元素,即局部递增的条件
#此时已经满足第i个元素>第i-1个元素了
if A[i] > A[i-1] and B[i] > B[i-1]:
#如果不换,那第i-1个元素也不能换,否则就可能不满足局部条件
dp[i][0] = min(dp[i][0], dp[i-1][0])
#如果换了,那第i-1个元素也必须跟着换,才能满足局部条件,
# 所以在i-1的基础上加一,把第i元素换过去
dp[i][1] = min(dp[i][1], dp[i-1][1]+1)
#此时是都比对方的前一个元素大
if A[i] > B[i-1] and B[i] > A[i-1]:
#如果第i个元素不换,那就只有把第i-1个元素交换后才满足局部递增的条件
dp[i][0] = min(dp[i][0], dp[i-1][1])
#如果第i个元素要换,那就正好满足局部递增的条件,那第i-1个元素就不用交换,
dp[i][1] = min(dp[i][0], dp[i-1][0]+1)
print(min(dp[N][0], dp[N][1]))