C++刷题篇——01组合出合法最小数

一、题目

二、解题思路

解题关键在于使用一个函数实现字符串组合后值的最小排序，然后在考虑首字母为0的情况

用到的方法：

切割字符串放进数组、遍历数组、定义一个字符串拼接后转成整型降序排列的方法

三、代码

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>

using namespace std;

// 将输入的字符串从空格切割后放进数组
vector<string> split(string params) {
	vector<string>p;
	while (params.find(" ") != string::npos) {
		int found = params.find(" ");
		p.push_back(params.substr(0, found));
		params = params.substr(found + 1);
	}
	p.push_back(params);
	return p;
}

// 这个函数实现的是将：10、2、08、02、09、1排序为：02、08、09、10、1、2，即按这个排序后，组合成的字符串转成整型值最小
bool comp(string a, string b) {
	return stoi(a + b) < stoi(b + a); // 按组合后降序排列。注意此处是两个字符串相加后转成整型再比较
}


int main() {
	string nums_str;
	getline(cin, nums_str);
	vector<string>nums_vec = split(nums_str); // 将字符串切割后放进数组

	sort(nums_vec.begin(), nums_vec.end(), comp); // 遍历数组元素，按照comp定义的方法排序

	// 要考虑首位不能为0，所以如果首位为0，那么要遍历找到首位不为0的值作为首位
	if (nums_vec[0][0] == '0') {
		for (int i = 1; i < nums_vec.size(); i++) { // 注意从i=1开始遍历
			if (nums_vec[i][0] != '0') { // 如果数组i位的首字母不为0，说明满足条件
				nums_vec[0] = nums_vec[i] + nums_vec[0]; // 那么i位的字符串加上首位（为0）的字符串作为数组新的首位
				nums_vec[i] = ""; // i位使用完就要置空
				break; // 找到了就结束遍历
			}
		}
	}

	// 此时数组的首元素不为0了，且顺序也满足从小到大，仅需将各位字符串相加即可
	string res = "";
	for (int i = 0; i < nums_vec.size(); i++) {
		res += nums_vec[i];
	}

	cout << res;
	return 0;
}