腾讯九月6号红黑棋子顺序调整问题

记腾讯九月6号红黑棋子顺序调整问题。

对于这样做转移的最优性质还需要继续思考。

定义状态d[i][j]代表前i个黑子和前j个红子已经交换到了序列最前面且剩余棋子都按照原来顺序被交换到这i+j个棋子之后。

那么d[i][j] = max(d[i+1][j] + sum(i,j,black[i+1] -1) , d[i][j+1] + sum(i,j,red[j+1] -1))

sum(i,j,k)代表当前i个黑子和前j个红子已经交换到了序列最前面时，把剩余子中原来位置为K的子移动到i+j+1的位置，需要越过几个子，sum（i,j,k）可通过树桩数组实时维护。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn = 3000 + 12;

int d[maxn][maxn];

int n;
int c[maxn];
int lowbit(int x){
    return x&(-x);
}
void add(int i,int k){
    while(i <= n){
        c[i] += k;
        i += lowbit(i);
    }
}
int getsum(int i){
    int res = 0;
    while(i > 0){
        res += c[i];
        i -= lowbit(i);
    }
    return res;
}
string s;
int a[maxn],color[maxn],black[maxn],red[maxn];
void read_(){
    scanf("%d",&n);
    cin>>s;
    for(int i=1;i<=n+n;i++){
        cin>>a[i];
        if(s[i - 1] == 'B') {
            color[i] = 0;
            black[a[i]] = i;
        }else {
            color[i] = 1;
            red[a[i]] = i;
        }
    }
}
int main()
{
  read_();
  memset(d,n*n,sizeof(d));
  d[n][n] = 0;
  for(int i=n;i>=0;i--){
    for(int j=n;j>=0;j--){
       if(i !=n || j!=n){
          if(i<n) d[i][j] = min(d[i + 1][j]+getsum(black[i + 1]-1),d[i][j]);
          if(j<n) d[i][j] = min(d[i][j + 1]+getsum(red[j + 1]-1),d[i][j]);
       }
       if(j>0) add(red[j],1);
    }
    if(i>0) add(black[i],1);
    for(int j=n;j>=1;j--) add(red[j],-1);
  }
  cout<<d[0][0]<<endl;
  return 0;
}