1. 程式人生 > 實用技巧 >[PAT] A1038 Recover the Smallest Number

[PAT] A1038 Recover the Smallest Number

(貪心、字串)

題目大意

給一些字串,求它們拼接起來構成最小數字的方式

思路

貪心演算法。用cmp排序,直接判斷a+b和b+a的大小即可。
必須保證兩個字串構成的數字是最小的才行,所以cmp函式寫成return a + b < b + a;的形式,保證它排列按照能夠組成的最小數字的形式排列。
因為字串可能前面有0,這些要移除掉(用s.erase(s.begin())就可以了)。輸出拼接後的字串即可。
注意:如果移出了0之後發現s.length() == 0了,說明這個數是0,那麼要特別地輸出這個0,否則會什麼都不輸出。

AC程式碼

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<iostream>
#include<vector>
#include<queue>
#include<string.h>
#include<algorithm>
using namespace std;
#define MAX 10002
bool cmp(string a, string b) {
    return a + b < b + a;
}
int main() {
    string input[MAX];
    int i = 0;
    int num;
    cin >> num;
    for (i = 0;i < num;i++)
        cin >> input[i];
    sort(input, input + num, cmp);
    bool flag = 0;
    string ans;
    for (i = 0;i < num;i++)
        ans += input[i];
    while (ans.size() != 0 && ans[0] == '0')
        ans.erase(ans.begin());
    if (ans.size() == 0)
        cout << 0;
    else cout << ans;
    return 0;
}

AC程式碼2(複雜)

一開始我沒想到直接可以a+b拼接判斷,還考慮了多種情況。在處理首位0的時候也比較複雜。
要學會充分利用cmp和string的功能啊。。。。。。

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<iostream>
#include<vector>
#include<queue>
#include<string.h>
#include<algorithm>
using namespace std;
#define MAX 10002
bool cmp(string a, string b) {
    if (a.size() == b.size())return a < b;
    int i = 0, j = 0;
    while (1) {
        if (a[i] != b[j])
            return a[i] < b[j];
        i++;j++;
        if (i >= a.size() && j >= b.size())
            return 0;
        if (i >= a.size())
            i = i - a.size();
        if (j >= b.size())
            j = j - b.size();
    }

    //return a + b < b + a;

}
int main() {
    string input[MAX];
    int i = 0;
    int num;
    cin >> num;
    for (i = 0;i < num;i++)
        cin >> input[i];
    sort(input, input + num, cmp);
    bool flag = 0;
    for (i = 0;i < num;i++) {
        if (flag == 0) {
            for (int j = 0;j < input[i].size();j++) {
                if (flag==0 && input[i][j] != '0'){
                    flag = 1;
                    cout << (int)(input[i][j] - '0');
                }
                else if (flag == 1)
                    cout << (int)(input[i][j] - '0');
            }
        }
        else cout << input[i];
    }
    if (flag == 0)cout << 0;
    return 0;
}