bzoj 4881: [Lydsy2017年5月月赛]线段游戏 树状数组+set

题意

给出一个n的排列，求将该排列划分成两部分，使得其分别为上升序列的方案有多少种。答案模998244353.
n<=100000

分析

一开始yy出了一个dp方程，看形式应该是可以用数据结构来优化的。
实际上有更简单的方法。
首先把无解的情况判掉。若最长下降子序列大于2的无解。
我们可以在两两逆序对之间连边，那么答案就是2^联通块数量。
对于每个连通块，我们将其最大的元素仍到set里面。每加入一个元素，我们就把set里面大于它的连通块合并，再扔到set里面就好了。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>#include<set>using namespace std;typedef long long LL;const int N=100005;const int MOD=998244353;int n,a[N],f[N],c[N];set<int> w;void ins(int x,int y){    while (x<=n)    {        c[x]=max(c[x],y);        x+=x&(-x);    }}int find(int x){    int ans=0;    while (x)    {        ans=max(ans,c[x]);        x-=x&(-x);    }    return ans;}int ksm(int x,int y){    int ans=1;    while (y)    {        if (y&1) ans=(LL)ans*x%MOD;        x=(LL)x*x%MOD;y>>=1;    }    return ans;}int main(){    scanf("%d",&n);    for (int i=1;i<=n;i++) scanf("%d",&a[i]);    int mx=0;    for (int i=1;i<=n;i++)    {        f[i]=find(n-a[i])+1;        ins(n-a[i]+1,f[i]);        mx=max(mx,f[i]);    }    if (mx>2)    {        printf("0");        return 0;    }    for (int i=1;i<=n;i++)    {        int mx=a[i];        set<int>::iterator it=w.upper_bound(a[i]);        while (it!=w.end())        {            mx=max(mx,*it);            set<int>::iterator tmp=it;            it++;w.erase(tmp);        }        w.insert(mx);    }    printf("%d",ksm(2,w.size()));    return 0;}