题目大意:给你一个序列,叫你求最长上升子序列长度,但必须包含第k项。
解题思路:我们把k左边的比a[k]大的数去掉,k右边的比k小的数去掉,就可以保证选到a[k]了(因为左边的数小于a[k],而a[k]小于右边的数,取了a[k]一定比不取优)。
此题由于n到300000,需要优化后的LIS,我使用了单调队列优化。时间复杂度$O(n\log n)$。
C++ Code:
#include<cstdio> #include<algorithm> using namespace std; int n,k,a[300005],b[300005],q[300005]; int main(){ scanf("%d%d",&n,&k); for(int i=1;i<=n;++i) scanf("%d",&a[i]); int cnt=0; for(int i=1;i<k;++i) if(a[i]<a[k])b[++cnt]=a[i]; b[++cnt]=a[k]; for(int i=k+1;i<=n;++i) if(a[k]<a[i])b[++cnt]=a[i]; n=cnt; cnt=0; q[0]=-1; for(int i=1;i<=n;++i){ if(b[i]>q[cnt]){ q[++cnt]=b[i]; continue; } int p=lower_bound(q,q+cnt,b[i])-q-1; if(q[p+1]>b[i])q[p+1]=b[i]; } printf("%d\n",cnt); return 0; }
时间: 2024-12-18 23:54:08