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;}