一个数如果在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);
}