蓝桥杯模拟题-有条件的最长递增子序列-dp-python题解

蓝桥杯模拟题-j题

问题描述

给定 n 个整数 a[1], a[2], …, a[n]，小蓝希望在中间选出一部分，满足以下两个条件： 　　
1.对于某个下标集合 S，选出的数中有至少 k 个下标在集合 S 中； 　　
2. 选出的数按照原来的顺序排列，是严格单调上升的，即选出的是一个上升子序列。 　　
请问小蓝最多能选出多少个数？

输入格式

输入的第一行包含两个整数 n, k，用一个空格分隔。 　　第二行包含 n 个整数 a[1], a[2], …, a[n]，相邻的整数间用空格分隔。 　　第三行包含一个长度为 n 的01串，依次表示每个下标是否在集合 S 中，为 0 表示不在 S 中，为 1 表示在 S 中。

输出格式

输出一行包含一个整数，表示答案。如果没有满足条件的选法，输出-1。

样例输入

8 2

8 1 2 3 9 4 7 10

10001010

样例输出

3

样例说明

由于 8、9、7 三个数中至少要选 2 个，只能选 8 和 9，剩下的数只能选最后一个数 10。

样例输入

8 3

8 1 2 3 9 4 7 10

10001010

样例输出

-1

评测用例规模与约定

对于 30% 的评测用例，2 <= n <= 100，0 <= a[i] <= 100, 0 <= k <= 3。 对于所有评测用例，2 <= n <= 1000，0 <= a[i] <= 100000, 0 <= k <= 20。

采用动态规划求解，dp[i][j]表示从前i个数中取出j个在s中的数时，最长上升子序列的长度，初始化为-inf表示不可达，将dp[0][0]初始化为0
状态转移方程：
序号i在res中时：dp[i][j]=max(dp[p][j-1]+1,dp[i][j])
序号i不在res中：dp[i][j]=max(dp[p][j]+1,dp[i][j])
一般情况：dp[i][j]=max(dp[p][j],dp[i][j])

n,k=map(int,input().split())
a=[0]+list(map(int,input().split()))
s=list(input())

res=[]
for i in range(n):
    if s[i]=='1':
        res.append(i+1)

ini=-0x3f3f3f3f

dp=[[ini for i in range(n+1)]for j in range(n+1)]
dp[0][0]=0


for i in range(1,n+1):
    for j in range(0,i+1):
        if a[i]>min(a[:i]):
            if i in res:
                for p in range(i):
                    if a[i]>a[p]:
                        dp[i][j]=max(dp[p][j-1]+1,dp[i][j])

            else:
                for p in range(i):
                    if a[i]>a[p]:
                        dp[i][j]=max(dp[p][j]+1,dp[i][j])
        else:
            for p in range(i):
                dp[i][j]=max(dp[p][j],dp[i][j])
if dp[n][k]<0:
    print(-1)
else:
    print(dp[n][k])