51nod 1218 最长递增子序列 V2 [dp]

来源:互联网 发布:证件照片裁剪软件 编辑:程序博客网 时间:2024/06/05 07:52

一个数如果在lis中出现,位置是一定的(吧,反正这样过了…)。
正着搞一遍lis,反着搞一遍(最长递减子序列),数在lis里的位置就确定了
如果它在这两个序列中的位置加起来为(lis长度+1)就说明这个数在lis中出现
记录lis每一个位置可以放的数的个数,为1则一定有,大于1则可能有

#include<cstdio>#include<algorithm>#define N 50000using 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);}