题目
给定一个长度不超过 10^4
的、仅由英文字母构成的字符串。请将字符重新调整顺序,按 PATestPATest… 这样的顺序输出,并忽略其它字符。当然,六种字符的个数不一定是一样多的,若某种字符已经输出完,则余下的字符仍按 PATest 的顺序打印,直到所有字符都被输出。
输入格式:
输入在一行中给出一个长度不超过 10^4的、仅由英文字母构成的非空字符串。
输出格式:
在一行中按题目要求输出排序后的字符串。题目保证输出非空。
输入样例:
redlesPayBestPATTopTeePHPereatitAPPT
输出样例:
PATestPATestPTetPTePePee
思路
遍历字符串一遍,统计出PATest的个数,以及它们的总和。
然后依次循环输出PATest,每输出一个就减少该字符的存量,若没有了就跳过。当总和降为0时结束循环。
求数组和用到了numeric.h中的accumulate函数:
template< class InputIt, class T >
T accumulate( InputIt first, InputIt last, T init);
template< class InputIt, class T, class BinaryOperation >
T accumulate( InputIt first, InputIt last, T init,
BinaryOperation op );
其中第三个参数init是累加的初值,最后一个可选参数op是自定义回调函数。
代码
#include <iostream>
#include <vector>
#include <numeric>
using namespace std;
int main(){
string s;
cin >> s;
vector<int> nums(6);
for (int i=0; i<s.length(); i++){
switch(s[i]){
case 'P': nums[0]++;break;
case 'A': nums[1]++;break;
case 'T': nums[2]++;break;
case 'e': nums[3]++;break;
case 's': nums[4]++;break;
case 't': nums[5]++;break;
}
}
s = "PATest";
int count = accumulate(nums.begin(), nums.end(), 0);
int i = 0;
while(count!=0){
if (nums[i]!=0){
cout << s[i];
nums[i]--;
count--;
}
i = (i + 1) % 6;
}
return 0;
}