题解 | 字符串替换

#include <cstdio>
class StringFormat 
{
public:
    string formatString(string A, int n, vector<char> arg, int m) 
    {
        string ret;
        int idx = 0;
        for(int i = 0; i < A.size(); i++)
        {
            if(A[i] >= 'A' && A[i] <= 'Z') ret += A[i];
            else 
            {
                ret += arg[idx++];
                i++;
            }
        }
        if(idx < m)
            for(int i = idx; i < m; i++)
                ret += arg[i]; 
        return ret;
    }
};