大厂笔试真题网址:https://codefun2000.com/
塔子哥刷题网站博客:https://blog.codefun2000.com/
最多修改两个字符,生成字典序最小的回文串
提交网址:https://codefun2000.com/p/P1089
由于字符串经过修改一定为回文串,且最多修改两次,所以原字符串位置i
与对称位置n-i-1
不一样的个数最多为2。所以统计一下需要改的位置个数,记为cnt
- cnt=0,即原字符串就是回文串,找到第一个不为a的位置i,将
i
与对称位置n-i-1
都改为a
aca ---> aaa
acca ---> aaaa
acbca ---> aabaa
- cnt=1,只有一个位置需要修改,此时分两种情况
- 如果
i
与对称位置n-i-1
都不是a,将这俩位置都改为a即可 - 如果
i
与对称位置n-i-1
只有一个为a,将另一个不是a的改为a。此时只改了一个字母,还可以改一个。当字符串长度为奇数时,将中间字符改为a
abcdba ---> abaaba
abcea ---> aacaa
cbcac ---> caaac
- cnt=2,两个对称位置需要修改,
i
与对称位置n-i-1
,谁小就改为谁
abcde ---> abcba
int main() {
string s;
cin >> s;
int n = s.size();
vector<int> idxs;
for (int i = 0; i < n / 2; i++) {
if (s[i] != s[n - 1 - i]) {
idxs.push_back(i);
}
}
int cnt = idxs.size();
if (cnt == 0) {
// 本身就是回文串,需要找到第一个不是a的地方修改
for (int i = 0; i <= n / 2; i++) {
if (s[i] != 'a') {
s[i] = 'a';
s[n - 1 - i] = 'a';
break;
}
}
}
else if (cnt == 1) {
// 只有一个需要修改的地方
if (s[idxs[0]] != 'a' && s[n - 1 - idxs[0]] != 'a') {
// 如果对称位置都不是a,则两个都改为a
s[idxs[0]] = 'a';
s[n - 1 - idxs[0]] = 'a';
}
else {
// 如果只有一个为a,则修改不是a的那个
// 如果当前串为奇数,还要修改中间位置的为a
s[idxs[0]] = 'a';
s[n - 1 - idxs[0]] = 'a';
if (n & 1) {
s[n / 2] = 'a';
}
}
}
else {
// 有两个需要修改的地方,当前位置idxs[i]和对称位置n - 1 - idxs[i],谁小就改为谁
for (int i = 0; i < cnt; i++) {
char c = min(s[idxs[i]], s[n - 1 - idxs[i]]);
s[idxs[i]] = c;
s[n - 1 - idxs[i]] = c;
}
}
cout << s << endl;
return 0;
}