HDU 5125 Magic Balls DP(LIS变形)

HDU 5125
题意:给出长度n的序列a,b.操作:swap(a[i],b[i]) 问操作次数<=m次的情况下,序列a的LIS长度?
n,m<=1e3,a[i],b[i]<=1e9.

一般来说求LIS时 设d[i]为以a[i]结尾时的最长长度,现在第i为可能为a[i]或者b[i]加上一维.
d[i][0/1] 其前面一个 j 要么为a[j]或者b[j] 按题意转移即可
交换次数不超过m,设置num[i][0/1]表示到达该状态需要用到的最小交换次数.

若num[i][0/1]>m时 则d[i][0/1]要扣掉多余的交换次数,此时d[i][0/1]就是以i结尾交换次数不超过m时的LIS长度

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e3+5,inf=0x3f3f3f3f;
int n,m,a[N],b[N],d[N][2],c[N][2];
void update(int i,int j,int x,int y)
{
    d[i][x]=d[j][y]+1;
    c[i][x]=c[j][y];
    if(x)
        c[i][x]++;
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        memset(d,0,sizeof(d));
        memset(c,0,sizeof(c));
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            scanf("%d%d",&a[i],&b[i]);
        for(int i=1;i<=n;i++)
            for(int j=0;j<i;j++)
        {
            if(a[i]>a[j])
                if(d[i][0]<d[j][0]+1||(d[i][0]==d[j][0]+1&&c[i][0]>c[j][0]))
                    update(i,j,0,0);
            if(a[i]>b[j])
              if(d[i][0]<d[j][1]+1||(d[i][0]==d[j][1]+1&&c[i][0]>c[j][1]))
                    update(i,j,0,1);
            if(b[i]>a[j])
                if(d[i][1]<d[j][0]+1||(d[i][1]==d[j][0]+1&&c[i][1]>c[j][0]+1))
                    update(i,j,1,0);
            if(b[i]>b[j])
                if(d[i][1]<d[j][1]+1||(d[i][1]==d[j][1]+1&&c[i][1]>c[j][1]+1))
                    update(i,j,1,1);
        }
        int ans=1;
        for(int i=1;i<=n;i++)
        {
            if(m>=c[i][0])
                ans=max(ans,d[i][0]);
            else
                ans=max(ans,d[i][0]-c[i][0]+m);//swap only m times
            if(m>=c[i][1])
                ans=max(ans,d[i][1]);
            else
                ans=max(ans,d[i][1]-c[i][1]+m);
        }
        printf("%d\n",ans);
    }
    return 0;
}