AcWing 4653. 数位排序(vector<pair>+lambda表达式匿名函数)
原题链接
简单
作者:
V1Eerie
,
2023-01-03 18:10:05
,
所有人可见
,
阅读 213
C++ 代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<pair<int, int>> v;
int n;
int Nreturn(int n)
{
int sum = 0;
while (n)
{
sum += n % 10;
n /= 10;
}
return sum;
}
int main()
{
cin >> n;
for (int i = 1; i <= n; i++)
{
v.push_back(make_pair(i, Nreturn(i)));
}
sort(v.begin(), v.end(), [](pair<int, int> a, pair<int, int> b)
{
if (a.second == b.second)
return a.first < b.first;
else
return a.second < b.second; });
int a;
cin >> a;
cout << v[a - 1].first << endl;
return 0;
}