第13届蓝桥杯模拟赛题解最后一题

问题描述
　　给定 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来做。
下面是分析方法

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;
const int N=1010,M=22;
const int INF=2e9;
int f[N][M];
int n,m;
int a[N];
string s;
int main()
{
	cin>>n>>m;
	for(int i=0;i<n;i++)cin>>a[i];
	cin>>s;
	f[0][0]=1;
	f[0][1]=1;
	for(int k=2;k<=m;k++)
		f[0][k]=-INF;
	for(int i=0;i<n;i++)
		f[i][0]=1;
	for(int i=1;i<n;i++)
	{
		for(int k=0;k<=m;k++)
		{
			int temp=-INF;
			if(s[i]=='1')
			{
				for(int t=0;t<i;t++)
				if(a[t]<a[i])
				temp=max(temp,f[t][k-1]+1);
			}
			else
			{
				for(int t=0;t<i;t++)
				if(a[t]<a[i])
				temp=max(temp,f[t][k]+1);
			}
			f[i][k]=temp;
		}
	}
	if(f[n-1][m]<m)cout<<-1<<endl;
	else cout<<f[n-1][m]<<endl;
	return 0;
}