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

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