字符串替换

请你实现一个简单的字符串替换函数。原串中需要替换的占位符为"%s",请按照参数列表的顺序一一替换占位符。若参数列表的字符数大于占位符个数。则将剩下的参数字符添加到字符串的结尾。

给定一个字符串A，同时给定它的长度n及参数字符数组arg，请返回替换后的字符串。保证参数个数大于等于占位符个数。保证原串由大小写英文字母组成，同时长度小于等于500。

输入样例

"A%sC%sE",7,['B','D','F']

输出样例 

 返回："ABCDEF"

#include <iostream>
#include <stdio.h>
#include <string>
#include <vector>
using namespace std;

string formatString(string A, int n, vector<char> arg) {
    int count = 0;
    for (int i = 0; i < n-1; i++) {
        if (A[i] == '%' && A[i+1] == 's') {
            A[i] = arg[count];
            A.erase(i+1, 1);
            count++;
        }
    }
    if (count != arg.size()) {
        A += arg[count];
        count++;
    }
    return A;
}

int main()
{
    string A = "A%sC%sE";
    int n = 7;
    vector<char> arg;
    arg.push_back('B');
    arg.push_back('D');
    arg.push_back('F');
    A = formatString(A, n, arg);
    cout << A << endl;
    return 0;
}