【PAT】乙级 1043 输出PATest (20 分) c++

1043 输出PATest (20 分)

给定一个长度不超过 10⁴的、仅由英文字母构成的字符串。请将字符重新调整顺序，按 PATestPATest… 这样的顺序输出，并忽略其它字符。当然，六种字符的个数不一定是一样多的，若某种字符已经输出完，则余下的字符仍按 PATest 的顺序打印，直到所有字符都被输出。

输入格式：
输入在一行中给出一个长度不超过10⁴的、仅由英文字母构成的非空字符串。

输出格式：
在一行中按题目要求输出排序后的字符串。题目保证输出非空。

输入样例：

redlesPayBestPATTopTeePHPereatitAPPT

输出样例：

PATestPATestPTetPTePePee

示例代码：

#include<iostream>
#include<string>
using namespace std;
int num[6][2] = { 'P',0,'A',0,'T',0,'e',0,'s',0,'t',0 };
int main() {
	string str;
	cin >> str;
	for (int i = 0; i < str.size(); i++) {
		if (str[i] == 'P') {
			num[0][1]++;
		}
		if (str[i] == 'A') {
			num[1][1]++;
		}
		if (str[i] == 'T') {
			num[2][1]++;
		}
		if (str[i] == 'e') {
			num[3][1]++;
		}
		if (str[i] == 's') {
			num[4][1]++;
		}
		if (str[i] == 't') {
			num[5][1]++;
		}
	}
	for (int i = 0; num[0][1] + num[1][1] + num[2][1] + num[3][1] + num[4][1] + num[5][1] > 0; i++) {
		if (i == 6) {
			i = 0;
		}
		if (num[i][1] > 0) {
			cout << (char)num[i][0];
			num[i][1]--;
		}
	}
	return 0;
}