问题描述
LIS问题是最经典的动态规划基础问题之一。
如果要求一个满足一定条件的最长上升子序列,你还能解决吗?
给出一个长度为N整数序列,请求出它的包含第K个元素的最长上升子序列。
例如:对于长度为6的序列<2,7,3,4,8,5>,它的最长上升子序列为<2,3,4,5>,
但如果限制一定要包含第2个元素,那满足此要求的最长上升子序列只能是<2,7,8>了。
输入数据
第一行为两个整数N,K,如上所述。
接下来是N个整数,描述一个序列。
输出数据
请输出两个整数,即包含第K个元素的最长上升子序列长度。
样例输入与输出
8 6
65 158 170 299 300 155 207 389
3
数据范围
对于所有的数据,满足0 < n <= 200000,0 < k <= n
思路
题目都直接告诉你是DP了
直接将在 k 号元素之前的元素控制到只有比 a[k] 小的
之后的元素只有比他大的 a[k] 大的
然后分别求两个最长上升子序列,将所求的两个长度相加再加一
下面来证明一下
k前面的所有比 a[k] 大的元素如果在最长上升子序列里,那么 a[k]就不能取
与题设不符,后面的元素同理。所以为了有解,所有有他们存在的最长子序列都是错误的
如果我们不能取这些元素的话,就等价于把他们从原数组删掉
因为本身那个元素也在那个序列中,所以就要将最后的答案加一
最后因为题目数据范围是200000,那么用常规On^2的解法坑定要超时几个点
那么就需要用O nlogn 的解法,找后继替换
想知道详细证明过程的麻烦去看别的文章,这里简单解释一下
定义s数组,s[i]表示长度为i的最长上升子序列的结尾元素的值。
但是长度为i的最长上升子序列可能有很多个,应该保存哪个序列的结尾元素呢?
显然应该保存最小的那个(可以用反证法证明)。这样,s就是一个递增序列了
可以通过二分查找来获得一个可以更新当前状态的的最优值。这样算法就是nlogn的了。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,k,len,ans=1;
int a[200005],b[200005],c[200005];
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;++i) scanf("%d",&a[i]);
for(int i=1;i<k;++i){
if(a[i]>=a[k]) continue;//删除比a[k]大的元素
if(a[i]>b[len]) b[++len]=a[i];
else b[lower_bound(b,b+k,a[i])-b]=a[i];
//找后继,因为lower_bound取得是地址,减去首地址就是在数组中的位置
}
ans+=len,len=0;//更新答案,清零长度
for(int i=k+1;i<=n;++i){
if(a[i]<=a[k]) continue;//删除比a[k]小的元素
if(a[i]>b[len]) c[++len]=a[i];
else c[lower_bound(c,c+k,a[i])-c]=a[i];
}
ans+=len;
printf("%d\n",ans);
return 0;
}