bzoj 3688: 折线统计 （DP+树状数组）

题目描述

传送门

题解

将所有的点按照x排序
$f[i][j][0/1]$表示以第i个结尾，分成了j段，最后一段上升/下降的方案数。
然后用树状数组优化DP

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 100003
#define p 100007
using namespace std;
struct data{
    int x,y;
}a[N];
int f[N][12][2],g[12][2][N],n,k,mx;
int cmp(data a,data b)
{
    return a.x<b.x;
}
int lowbit(int x){
    return x&(-x);
}
void change(int tr[N],int x,int v){
    for (int i=x;i<=mx;i+=lowbit(i)) tr[i]=(tr[i]+v)%p;
}
int query(int tr[N],int x)
{
    int ans=0;
    if (!x) return 0;
    for (int i=x;i>=1;i-=lowbit(i)) ans=(ans+tr[i])%p;
    return ans;
}
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++) scanf("%d%d",&a[i].x,&a[i].y),mx=max(mx,a[i].y);
    sort(a+1,a+n+1,cmp);
    int ans=0;
    for (int i=1;i<=n;i++){//0上升,1下降 
       f[i][0][0]=f[i][0][1]=1;
       int v=a[i].y; 
       change(g[0][0],v,1); change(g[0][1],v,1);
       for (int j=1;j<=k;j++) {
            f[i][j][0]+=query(g[j][0],v-1)+query(g[j-1][1],v-1);
            f[i][j][1]+=query(g[j][1],mx)-query(g[j][1],v)+query(g[j-1][0],mx)-query(g[j-1][0],v);
            f[i][j][0]%=p;
            f[i][j][1]=(f[i][j][1]%p+p)%p;
            change(g[j][0],v,f[i][j][0]);
            change(g[j][1],v,f[i][j][1]);
       }
       ans=(ans+f[i][k][0]+f[i][k][1])%p;
    }
    printf("%d\n",ans);
}