排序
对于1e6
的数据范围,可以使用sort
排序
将 {数位之和,原数组} 存成二元组,排序输出第k
个
二元组第一位是位数和,第二个是原数字
sort对pair类型数组排序,以first
为第一关键字,如果第first
相同就按second
排序。
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<int, int> PII;
int main()
{
vector<PII> q;
int n, k;
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i ++ )
{
int sum = 0;
// 统计数位和
for (int x = i; x; x /= 10)
sum += x % 10;
q.push_back({sum, i});
}
sort(q.begin(), q.end());
printf("%d", q[k - 1].second);
}