CF339A - Helpful Maths

CF339A - Helpful Maths

题目

Helpful Maths - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

思路

• 题目要把1+3+4+2变成1+2+3+4这种有序的加法，显然只要把数字排序即可。
• 我们可以在初始字符串的末尾加上一个‘+’号，1+3+4+2 => 1+3+4+2+。
• 按‘+’把字符串分割可得，1+，3+，4+，2+，再从小到大进行排序可得1+，2+，3+，4+。
• 可以利用双指针分割字符串，再调用系统sort()函数进行排序（默认从小到大），
• 显然我们把排序后的字符串一个个加起来再去掉末尾的‘+’号即可得到答案。

代码

#include <bits/stdc++.h>
using namespace std;
// 定义数组容量
const int N = 107;

// 存放初始字符串
string str;
// 存放分割后的字符串
string strs[N];

int main() {
    cin >> str;
    // 在初始字符串末尾加上一个‘+’号，减少判断
    str += '+';
    // 记录分割了几个字符串
    int cnt = 0;
    // 按‘+’号分割字符串
    for (int i = 0, j = 0; j < str.size(); i++, j++) {
        // 将右指针移到‘+’的位置
        while (str[j] != '+') {
            j++;
        }
        // 保存分割的字符串[i, j]长度为j - i + 1
        strs[cnt++] = str.substr(i, j - i + 1);
        // 将左指针起点设为j, 循环会自动加一到下一个字符起点
        i = j;
    }
    // 利用库函数sort()进行排序，起始地址strs，结束地址strs + cnt
    sort(strs, strs + cnt);
    // 初始化答案为空字符串
    string ans = "";
    for (int i = 0; i < cnt; i++) {
        // 累加分割出的子串
        ans += strs[i];
    }
    // 将末尾的‘+’号去掉，并输出答案
    cout << ans.substr(0, ans.size() - 1) << endl;
    return 0;
}

复杂度分析

• 时间复杂度O(nlogn)
• 空间复杂度O(n)