如何自定義sort函式中的比較函式
題目描述
輸入一個正整數陣列,把數組裡所有數字拼接起來排成一個數,列印能拼接出的所有數字中最小的一個。例如輸入陣列{3,32,321},則打印出這三個數字能排成的最小數字為321323。
思路:自定義比較器,若a+b>b+a則a>b,即”3”+”23”>”23”+”3”則3>23,並且我們希望在排序的時候將23排在3的前面,也就是升序排列。
class Solution { public: static bool compare(const string& s1, const string& s2) { string ab = s1 + s2; string ba = s2 + s1; return ab < ba; //升序排列。如改為ab > ba, 則為降序排列 } string PrintMinNumber(vector<int> numbers) { string result; if(numbers.size() <= 0) return result; vector<string> num2str; for(int i = 0; i < numbers.size(); i++) { stringstream ss; ss << numbers[i]; string s = ss.str(); num2str.push_back(s); } sort(num2str.begin(), num2str.end(), compare); for(int i = 0; i < num2str.size(); i++) { result.append(num2str[i]); } return result; } };
這道題的解法中用到了自定義比較器。也就是自定義了campare函式。
但是為什麼當ab<ba的時候就是升序排列了呢?十分不解,在網上搜了好多資料,最後看到了c++的技術文件,醍醐灌頂!!!強烈推薦直接看技術文件,比在網上查來查去明白的更快!
// sort algorithm example #include <iostream> // std::cout #include <algorithm> // std::sort #include <vector> // std::vector bool myfunction (int i,int j) { return (i<j); } struct myclass { bool operator() (int i,int j) { return (i<j);} } myobject; int main () { int myints[] = {32,71,12,45,26,80,53,33}; std::vector<int> myvector (myints, myints+8); // 32 71 12 45 26 80 53 33 // using default comparison (operator <): std::sort (myvector.begin(), myvector.begin()+4); //(12 32 45 71)26 80 53 33 // using function as comp std::sort (myvector.begin()+4, myvector.end(), myfunction); // 12 32 45 71(26 33 53 80) // using object as comp std::sort (myvector.begin(), myvector.end(), myobject); //(12 26 32 33 45 53 71 80) // print out content: std::cout << "myvector contains:"; for (std::vector<int>::iterator it=myvector.begin(); it!=myvector.end(); ++it) std::cout << ' ' << *it; std::cout << '\n'; return 0; }
可以看到comp的定義:comp函式返回一個bool型別的值,這個值表示了在嚴格弱排序中(可以理解為升序排序)第一引數是否位於第二個引數之前。
也就是說如果comp返回true,則第一個引數小於第二個引數,sort根據compare的返回值將第一個引數排在第二個引數之前。
如果comp返回false,則第一個引數大於第二個引數,sort根據compare的返回值將第一個引數排在第二個引數之後。
傳入A,B,定義bool myfunction (int i,int j) { return (i<j); },作為comp函式,則排列AB。
傳入BA,則排列AB。
可以看出是升序排列的。(sort函式預設的comp函式也是預設升序的)
而如果我們定義bool myfunction (int i,int j) { return (i>j); },作為comp函式,
傳入AB,返回false,則排列為BA,
傳入BA,返回true,排列為BA。
是降序排列的。
回到最初的問題中:
static bool compare(const string& s1, const string& s2)
{
string ab = s1 + s2;
string ba = s2 + s1;
return ab < ba; //升序排列。如改為ab > ba, 則為降序排列
}
我們可以看出 如果s1=”3”, s2=”23”
ab = “323”;
ba = “233”;
事實上ab>ba,所以comp會返回false,sort根據compare返回的false將s2排列在s1之前,也就是排列成”23”,”3”這也就是我們想要的結果。
總結起來就是:
sort函式根據comp函式的返回值,對comp函式的兩個引數排序。
如果comp返回true,排序為“引數1”“引數2”,否則排序為“引數2”“引數1”。
想要升序排列,則return parameter1<parameter2
想要降序排列,則return parameter1>parameter2