PAT乙级真题 1043 输出PATest C++实现

题目

给定一个长度不超过 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;
}