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
);
return 0;
}
else if(k==1)
{memset(s,-127/3,sizeof(s));
s[1]=a
; 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
; 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;
}