hdu 5125 magic balls LIS (dp+树状数组优化）

昨天BC的第三题，一开始用线段树写的，先是MLE，全改成short之后一直TLE到比赛结束，后来听别人说树状数组优化能过，后来试了一下果然能够，坑啊（以后LIS用树状数组了，比线段树快的多）

至于树状数组求1-n区间最大值和求和差不多思想。

dp转移方程是：

dp[i][j][0]=max(dp[k][j][0]+1(a[k]<a[i]),dp[k][j][1]+1(b[k]<a[i]));

dp[i][j][1]=max(dp[k][j-1][0]+1(b[i]>a[k]),dp[k][j-1][1]+1(b[k]<b[i]));

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
typedef short sh;
int low(int a){return a&(-a);}
int bit[1005][2][1005];
void merg(int p,int k,int f,int n,int v){
    while(f<=n){
        bit[p][k][f]=max(bit[p][k][f],v);
        f=f+low(f);
    }
}
int query(int p,int k,int n){
    int s=0;
    while(n>0){
        s=max(s,bit[p][k][n]);
        n=n-low(n);
    }
    return s;
}
int a[1005];
int b[1005];
int dp[1005][1005][2];
struct pppi{
    int a;
    int b;
}pp2[1005];
int main()
{
    int i,j,n,m,p,t,p1,x1,y1,x2,y2;
    int k;
    cin>>t;
    while(t--){
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++){
            scanf("%d%d",&pp2[i].a,&pp2[i].b);
            a[i]=pp2[i].a;
            b[i]=pp2[i].b;
        }
        sort(a+1,a+1+n);
        sort(b+1,b+1+n);
        memset(bit,0,sizeof(bit));
        memset(dp,0,sizeof(dp));
        for(i=0;i<=m;i++){
            dp[1][i][0]=1;
            if(i>0) dp[1][i][1]=1;
            p=lower_bound(a+1, a+1+n, pp2[1].a)-a;
            merg(i,0,p,n,1);
            if(i>0){
            p=lower_bound(b+1, b+1+n, pp2[1].b)- b;
            merg(i,1,p,n,1);
            }
        }
        k=1;
        for(i=2;i<=n;i++){
            p=lower_bound(a+1, a+1+n, pp2[i].a)-a;
            p1=lower_bound(b+1, b+1+n, pp2[i].b)-b;
            x1=lower_bound(a+1, a+1+n, pp2[i].a-1)-a;
            if(x1>n||a[x1]>pp2[i].a-1) x1--;
            y1=lower_bound(b+1, b+1+n, pp2[i].b-1)-b;
            if(y1>n||b[y1]>pp2[i].b-1) y1--;
            x2=lower_bound(a+1, a+1+n, pp2[i].b-1)-a;
            if(x2>n||a[x2]>pp2[i].b-1) x2--;
            y2=lower_bound(b+1, b+1+n, pp2[i].a-1)-b;
            if(y2>n||b[y2]>pp2[i].a-1) y2--;
            for(j=m;j>=0;j--){
                dp[i][j][0]=query(j,0,x1)+1;
                dp[i][j][0]=max(dp[i][j][0],query(j,1,y2)+1);
                if(j>0){
                    dp[i][j][1]=query(j-1,1,y1)+1;
                    dp[i][j][1]=max(dp[i][j][1],query(j-1,0,x2)+1);
                }
                k=max(k,dp[i][j][1]);
                k=max(dp[i][j][0],k);
                merg(j,0,p,n,dp[i][j][0]);
                merg(j,1,p1,n,dp[i][j][1]);
            }
        }
        printf("%d\n",k);
    }
}