[PAT] A1038 Recover the Smallest Number
阿新 • • 發佈:2020-07-13
(貪心、字串)
題目大意
給一些字串,求它們拼接起來構成最小數字的方式
思路
貪心演算法。用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; }