Codeforces Round #585 (Div. 2)E. Marbles（状压dp）

https://codeforces.com/contest/1215/problem/E

cnt[x][j]表示把x这种颜色全放到j这种颜色前面所需要的交换次数。

其实这题跟上一篇博客那题是同一个套路，难点在于想到预处理出cnt数组来进行转移。

#define bug(x) cout<<#x<<" is "<<x<<endl
#define IO std::ios::sync_with_stdio(0)
#include <bits/stdc++.h>
#define iter ::iterator
//#define pa pair<int,int>
using namespace  std;
#define ll long long
#define mk make_pair
#define pb push_back
#define se second
#define fi first
#define ls o<<1
#define rs o<<1|1
ll mod=998244353;
const int inf=2e9+10;
const int N=(1<<20)+5;
ll d[N],cnt[25][25],sum[25];
int n;
int main(){
    IO;
    cin>>n;
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        x--;
        for(int j=0;j<20;j++){
            cnt[x][j]+=sum[j];
        }
        sum[x]++;
    }
    for(int i=0;i<=N-2;i++)d[i]=1e18;
    d[0]=0;
    for(int s=0;s<(1<<20);s++){
        for(int i=0;i<20;i++){
            if(!((s>>i)&1)){
                ll res=0;
                for(int j=0;j<20;j++){
                    if((s>>j)&1){
                        res+=cnt[i][j];
                    }
                }
                int ns=s^(1<<i);
                d[ns]=min(d[ns],d[s]+res);
            }
        }
    }
    cout<<d[(1<<20)-1]<<endl;
}