A-Ant Typing
![](https://i-blog.csdnimg.cn/blog_migrate/a0333d0385f0302d096749c4613914cb.png)
一开始我就想到用枚举,但是估算错了枚举的规模,算成了9^9,觉得枚举规模过大就放弃了。实际上只有9!种情况,大概三十多万种,也就是A(9,9)。stl中提供了一个生成字典序的函数next_permutation(),可以用这个函数来简化代码量。对于计算结果,无需每次遍历输入的字符串,只要记录从一个数字到另一个数字的次数即可,也就是cnt[i][j]表示从i到j的次数,然后根据生成的字典序列计算就好。
#include <bits/stdc++.h>
using namespace std;
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
int cnt[10][10] = {0}; //cnt[i][j]记录蚂蚁从i+1到j+1的次数
int ans = 100000000, temp;
string str;
vector<int> perm(10,0);
cin >> str;
for (int i = 1; i < str.size(); i++)//统计cnt
{
cnt[str[i - 1] - '0'][str[i] - '0']++;
}
for (int i = 1; i <= 9; i++) //perm[i]表示数字i在第perm[i]个位置
{
perm[i]=i;
}
do
{
temp = perm[str[0]-'0']-1;//蚂蚁一开始位于最左边,加上从最左边到第一个数字的距离
for (int i = 1; i <= 9; i++)
{
for (int j = i + 1; j <= 9; j++)
{
temp += (cnt[i][j] + cnt[j][i]) * abs(perm[i]-perm[j]);
}
}
if (temp < ans)
{
ans = temp;
}
} while (next_permutation(perm.begin()+1, perm.end()));
ans += str.size();//每个位置都要停留一秒来敲击键盘
cout << ans ;
return 0;
}