codeforces1238E Keyboard Purchase

题目链接

题目：

给定一个长度为$n$的由前$m$个小写字母组成的字符串$s$，要你给出一个长度为$m$的一维键盘（前$m$个小写字母中的每一个字符占一个键），定义一个键盘的$slowness$值为${\sum }_{i=2}^n|po{s}_{s_i}-po{s}_{s_{i-1}}|$，其中$po{s}_c$为字符$c$在键盘中的位置。问能达到的最小的$slowness$值为多少。
$(1\le n\le 10^5,1\le m\le 20)$

题解：

字符集的范围不是26而是20就提醒我们要用状压DP了。
首先$slowness$值定义里有绝对值不好算贡献，所以将其去掉，那么对于每种字符$c$，会有两种贡献，一种是与比$po{s}_c$小的字符产生的，这种情况下字符$c$单独产生的贡献为$po{s}_c$；另一种是与比$po{s}_c$大的字符产生的，这种情况下字符$c$单独产生的贡献为$-po{s}_c$ 。更形式化的说，令 n e i g h b o r c = { p ∣ s p ≠ c ∧ ( s p − 1 = c ∨ s p + 1 = c ) } neighbor_c=\{p|s_p \ne c \land (s_{p-1}=c \lor s_{p+1}=c) \} neighborc​={p∣sp​​=c∧(sp−1​=c∨sp+1​=c)} ，字符$c$的贡献为${\sum }_{p\in neighbo{r}_c\wedge po{s}_{s_p}<po{s}_c}po{s}_c-{\sum }_{p\in neighbo{r}_c\wedge po{s}_{s_p}>po{s}_c}po{s}_c$
定义$d{p}_{msk}$为确定了键盘上的前$|msk|$个位置的字符，字符集为$msk$时这些确定位置的字符对$slowness$值产生的最小贡献。那么转移方程为
$$d{p}_{msk}=mi{n}_{x\in msk}(d{p}_{msk-x}+\sum _{p\in neighbo{r}_c\wedge s_p\in msk}po{s}_x-\sum _{p\in neighbo{r}_c\wedge s_p\notin msk}po{s}_x)$$
我们可以$O(n)$预处理出$a_{ch1,ch2}={\sum }_{p\in neighbo{r}_{ch1}\wedge s_p=ch2}$，然后$O(2^m\times m)$预处理出$cos{t}_{msk,c}={\sum }_{x\in msk}{a}_{x,c}$，那么转移方程可以变为
$$d{p}_{msk}=mi{n}_{x\in msk}(d{p}_{msk-x}+cos{t}_{msk-x,x}\times po{s}_x-(|neighbo{r}_x|-cos{t}_{msk-x,x})\times po{s}_x)$$
可以实现$O(1)$的转移
由于从小到大枚举$msk$的时候可以保证$msk$的子集比$msk$先枚举到，所以$dp$顺序的正确性可以保证。
关于预处理出$cos{t}_{msk,c}$，一种方式是枚举完$msk$和$c$之后，再枚举一遍$msk$中的元素$x$累加$a_{x,c}$算出结果，这样复杂度是$O(m^2\times 2^m)$，另一种方式是用利用子集计算出的结果进行递推，具体的来说，先处理出每一个$msk$中最小的元素$minbi{t}_{msk}$（其实不是最小的也行），然后$cos{t}_{msk,c}=cos{t}_{msk-minbi{t}_{msk},c}+a_{minbi{t}_{msk},c}$，这种方式的复杂度为$O(m\times 2^m)$

复杂度：$O(n+m\times 2^m)$

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<string>
#include<bitset>
#include<sstream>
#include<ctime>
//#include<chrono>
//#include<random>
//#include<unordered_map>
using namespace std;


#define ll long long
#define ls o<<1
#define rs o<<1|1
#define pii pair<int,int>
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define sz(x) (int)(x).size()
#define all(x) (x).begin(),(x).end()
const double pi=acos(-1.0);
const double eps=1e-6;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const int maxn=1e5+5;
const int maxm=20;
int n,m;
int dp[1<<maxm],a[maxm][maxm],cnt[maxm],b[1<<maxm][maxm],minbit[1<<maxm],numbit[1<<maxm];
char s[maxn];
int main(void){
	// freopen("in.txt","r",stdin);
	scanf("%d%d%s",&n,&m,s+1);
	for(int i=1;i<=n;i++){
		if(i!=1&&s[i]!=s[i-1]){
			a[s[i]-'a'][s[i-1]-'a']++;
			cnt[s[i]-'a']++;
		}
		if(i!=n&&s[i]!=s[i+1]){
			a[s[i]-'a'][s[i+1]-'a']++;
			cnt[s[i]-'a']++;
		}
	}
	for(int i=1;i<(1<<m);i++){
		minbit[i]=-1;
		for(int j=0;j<m;j++){
			if((i>>j)&1){
				if(minbit[i]==-1)minbit[i]=j;
				numbit[i]++;
			}
		}
	}
	for(int i=1;i<(1<<m);i++){
		for(int j=0;j<m;j++){
			int c=minbit[i];
			b[i][j]+=b[i^(1<<c)][j]+a[c][j];
		}
	}
	memset(dp,INF,sizeof(dp));
	dp[0]=0;
	for(int i=1;i<(1<<m);i++){
		for(int j=0;j<m;j++){
			if((i>>j)&1){
				int t=i^(1<<j);
				int v=dp[t]+b[t][j]*numbit[i]-(cnt[j]-b[t][j])*numbit[i];
				dp[i]=min(dp[i],v);
			}
		}
	}
	printf("%d\n",dp[(1<<m)-1]);
	return 0;
}