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