执行操作后字典序最小的字符串
题目
数据
- 字符串s,s长度为偶数
- 2个整数a,b。
操作
- 累加:s[i]+=a,i为所有奇数。(累加后不进位,相当于个位数相加保留个位数)。所有奇数下标都必须同时加上a
- 轮转(循环右移):s>>b。(1<=b<=s.length-1)
代码
class Solution {
public:
string findLexSmallestString(string s, int a, int b) {
int n = s.size(); //b<=n-1
vector<int> vis(n);
string res = s; //题目规定是偶数
// 将 s 延长一倍,方便截取轮转后的字符串 t
s = s + s;
for (int i = 0; vis[i] == 0; i = (i + b) % n) {;
vis[i] = 1;
for (int j = 0; j < 10; j++) {
int k_limit = b % 2 == 0 ? 0 : 9;//测验b是奇数还是偶数。奇数则k_limit=9,偶数则k_limit=0。
for (int k = 0; k <= k_limit; k++) {//如果b是偶数的话,则只循环一次
// 每次进行累加之前,重新截取t
string t = s.substr(i, n);//
for (int p = 1; p < n; p += 2) {//
t[p] = '0' + (t[p] - '0' + j * a) % 10;//
}//
for (int p = 0; p < n; p += 2) {//
t[p] = '0' + (t[p] - '0' + k * a) % 10;//
}
res = min(res, t);
}
}
}
return res;
}
};
第9,10行代码解释如下:
假设b=2,每次轮转都会把最末两位移动得到最前两位。我们跟踪字符串中的一个字符s[i],初始时i=0,s[i]=0。
然后轮转一次后i=(i+b)%n=2。且置vis[i]=1,表明i=2的情况出现过一次,如果再次出现i=2的情况则说明枚举了所有可能的情况了,此时应该停止循环。
如下图:
在个位数累加某个固定的数,必定会会有循环,当累加了某个固定的数10次后就会从头开始。因此设定j<10。
for (int j = 0; j < 10; j++) {
int k_limit = b % 2 == 0 ? 0 : 9;//测验b是奇数还是偶数
for (int k = 0; k <= k_limit; k++) {
// 每次进行累加之前,重新截取 t
string t = s.substr(i, n);
for (int p = 1; p < n; p += 2) {
t[p] = '0' + (t[p] - '0' + j * a) % 10;
}
for (int p = 0; p < n; p += 2) {
t[p] = '0' + (t[p] - '0' + k * a) % 10;
}
res = min(res, t);
}
}
如果轮转的字符个数为偶数的话,那么会被累加的字符是固定的,也即是奇数下标不会因为轮转而变成偶数下标。因此只需要计算奇数下标所有累加后的情况中的最小值。
如果轮转的字符个数为奇数的话,那么所有下标的字符都有可能因为轮转后成为奇数下标。因此需要考虑所有的情况。
第6-8行代码:这里将所有奇数下标的字符都累加一次。
第9-11行代码:这里将所有偶数下标的字符都累加一次。
然后保留最小值。