题目描述 Description
LIS问题是最经典的动态规划基础问题之一。如果要求一个满足一定条件的最长上升子序列,你还能解决吗?
给出一个长度为N整数序列,请求出它的包含第K个元素的最长上升子序列。
例如:对于长度为6的序列<2,7,3,4,8,5>,它的最长上升子序列为<2,3,4,5>,但如果限制一定要包含第2个元素,那么满足此要求的最长上升子序列就只能是<2,7,8>了。
输入描述 Input Description
第一行为两个整数N,K,如上所述。
接下来是N个整数,描述一个序列。
输出描述 Output Description
请输出两个整数,即包含第K个元素的最长上升子序列长度。
样例输入 Sample Input
8 6
65 158 170 299 300 155 207 389
样例输出 Sample Output
4
数据范围及提示 Data Size & Hint
80%的数据,满足0<n<=1000,0<k<=n
100%的数据,满足0<n<=200000,0<k<=n
题解
O(nlogn)的经典裸题,二分查找,但我的代码好像不漂亮……#include<cstdio> #include<iostream> #include<cstring> #include<cstdlib> #include<cmath> #include<algorithm> using namespace std; int n,k,l,r,da,f[200002],s[200002],a[200002]; void erf1(int z,int y,int w) { int ans=0,mid; while(z<=y) {mid=(z+y)>>1; if(s[mid]<a[w]) {z=mid+1; ans=mid;} else y=mid-1; } s[ans+1]=min(s[ans+1],a[w]); f[w]=ans+1; if(ans==r) r++; } void erf2(int z,int y,int w) { int ans=0,mid; while(z<=y) {mid=(z+y)>>1; if(s[mid]>a[w]) {z=mid+1; ans=mid;} else y=mid-1; } s[ans+1]=max(s[ans+1],a[w]); f[w]=ans+1; if(ans==r) r++; } int main() { scanf("%d%d",&n,&k); for(int i=1;i<=n;i++) scanf("%d",&a[i]); if(k==n) {memset(s,127/3,sizeof(s)); s[1]=a[1]; l=1; r=1; for(int i=2;i<=n;i++) erf1(l,r,i); //for(int i=1;i<=n;i++) printf("%d ",f[i]); printf("%d",f[n]); return 0; } else if(k==1) {memset(s,-127/3,sizeof(s)); s[1]=a[n]; l=1; r=1; for(int i=n-1;i>=1;i--) erf2(l,r,i); //for(int i=1;i<=n;i++) printf("%d ",f[i]); printf("%d",f[1]); return 0; } else {memset(s,127/3,sizeof(s)); s[1]=a[1]; l=1; r=1; for(int i=2;i<=k;i++) erf1(l,r,i); da=f[k]; //for(int i=1;i<=n;i++) printf("%d ",f[i]); //printf("\n%d\n",da); memset(s,-127/3,sizeof(s)); s[1]=a[n]; l=1; r=1; for(int i=n-1;i>=k;i--) erf2(l,r,i); //printf("%d\n",f[k]); da=da+f[k]-1; printf("%d",da); } return 0; }