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

                                                     E. Marbles

题意：

           给定n(4e5)个1到20的数，求最少经过几次相邻交换，使相同的数在一段中。

题解：

          考虑到ai只有20，可用状压dp来做，dp[0]表示什么都没有，dp状态转移按照数字出现的顺序转移，如加入数x，

dp[i|(1<<x-1)]=min(dp[i|(1<<x-1)],dp[i]+cnt[i中有1的位+1][x]).其中cnt[i][j]表示i排在j前面，只考虑两个数，产生的贡献是多少。

如此便做完了。

代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=4e5+9;
int n,a[N],g[29];
ll dp[(1<<20)+9],cnt[29][29];
int main(){
   // freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];g[a[i]]++;
        for(int j=1;j<=20;j++)if(a[i]!=j)cnt[a[i]][j]+=g[j];
    }
  //  for(int i=1;i<=4;i++)for(int j=1;j<=4;j++)cout<<cnt[i][j]<<endl;
    memset(dp,0x3f,sizeof(dp));
    dp[0]=0;
    int i;
    for(i=0;i<(1<<20);i++){
        for(int j=1;j<=20;j++){
            if(i&(1<<j-1))continue;
            ll sum=dp[i];
            for(int k=1;k<=20;++k)if(i&(1<<k-1))sum+=cnt[k][j];
            dp[i|(1<<j-1)]=min(dp[i|(1<<j-1)],sum);
        }
    }
    cout<<dp[i-1]<<endl;
    return 0;
}