wikioi 最长上升子序列题目

题目描述 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;
}