51nod 1218 最長遞增子序列 V2 [dp]
阿新 • • 發佈:2019-02-20
一個數如果在lis中出現,位置是一定的(吧,反正這樣過了…)。
正著搞一遍lis,反著搞一遍(最長遞減子序列),數在lis裡的位置就確定了
如果它在這兩個序列中的位置加起來為(lis長度+1)就說明這個數在lis中出現
記錄lis每一個位置可以放的數的個數,為1則一定有,大於1則可能有
#include<cstdio>
#include<algorithm>
#define N 50000
using namespace std;
inline int read()
{
int a=0,f=1;char c=getchar();
while(c<'0' ||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){a=a*10+c-'0';c=getchar();}
return a*f;
}
int n,a[N+5],dp[N+5],len,pre[N+5],suf[N+5],mark[2][N+5];
int main()
{
n=read();
int i,pos;
for(i=1;i<=n;++i) a[i]=read();
len=1,dp[len]=a[1];pre[1]=len;
for (i=2;i<=n;++i)
{
if(a[i]>dp[len]) dp[++len]=a[i],pre[i]=len;
else
{
pos=lower_bound(dp+1,dp+len,a[i])-dp;
pre[i]=pos;
dp[pos]=a[i];
}
}
len=1,dp[len]=-a[n];suf[n]=len;
for(i=n-1;i>=1;--i)
{
if(-a[i]>dp[len]) dp[++len]=-a[i],suf[i]=len;
else
{
pos=lower_bound(dp+1,dp+len,-a[i])-dp;
suf[i]=pos;
dp[pos]=-a[i];
}
}
for(i=1;i<=n;++i) //printf("%d %d\n",pre[i],suf[i]);
if(pre[i]+suf[i]==len+1) ++mark[0][i],++mark[1][pre[i]];
for(i=1,printf("A:");i<=n;++i)
if(mark[0][i]&&mark[1][pre[i]]>1) printf("%d ",i);
printf("\n");
for(i=1,printf("B:");i<=n;++i)
if(mark[0][i]&&mark[1][pre[i]]==1) printf("%d ",i);
}