AcWing 4653. 数位排序

AcWing 4653. 数位排序

原题

原题链接

小蓝对一个数的数位之和很感兴趣，今天他要按照数位之和给数排序。

当两个数各个数位之和不同时，将数位和较小的排在前面，当数位之和相等时，将数值小的排在前面。

例如，20222022 排在 409409 前面，因为 20222022 的数位之和是 66，小于 409409 的数位之和 1313。

又如，66 排在 20222022 前面，因为它们的数位之和相同，而 66 小于 20222022。

给定正整数 n，m�，�，请问对 11 到 n� 采用这种方法排序时，排在第 m� 个的元素是多少？

输入格式

输入第一行包含一个正整数 n�。

第二行包含一个正整数 m�。

输出格式

输出一行包含一个整数，表示答案。

数据范围

对于 30%30% 的评测用例，1≤m≤n≤3001≤�≤�≤300。
对于 50%50% 的评测用例，1≤m≤n≤10001≤�≤�≤1000。
对于所有评测用例，1≤m≤n≤1061≤�≤�≤106。

输入样例：

13
5

输出样例：

3

样例解释

11 到 1313 的排序为：1,10,2,11,3,12,4,13,5,6,7,8,91,10,2,11,3,12,4,13,5,6,7,8,9。

第 55 个数为 33。

题解

// 会出现TLE的代码
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
bool cmp(int s,int b){
	int ss=0,bs=0,st=s,bt=b;
	while(st>0)
	{
		ss+=st%10;
		st=st/10;
	}
	while(bt>0)
	{
		bs+=bt%10;
		bt=bt/10;
	}
	if(bs==ss) return s<b;
	return ss<bs;	
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	int nums[n+1];
	for(int i=1;i<=n;i++)
		nums[i]=i;
	sort(nums+1,nums+n+1,cmp);
	printf("%d",nums[m]);
	return 0;
}

上面是会出现tle的代码，快排的时间复杂度达到O（nlogn），假设大部分的数都有6位，则达到了大概108，所以不能不能将处理数位放到排序里面乘，所以改成下面的：

#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
const int N=1000010;
int nums[N],s[N]={0};
bool cmp(int small,int big){
	if(s[small]==s[big]) return small<big;
	return s[small]<s[big];	
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		nums[i]=i;
		int bt=nums[i];
		s[i]=0;
		while(bt>0)
		{
			s[i]+=bt%10;
			bt=bt/10;
		}	
	}	
	sort(nums+1,nums+n+1,cmp);
	printf("%d",nums[m]);
	return 0;
}

• sort函数是c++封装的一个函数（可以自动选择排序算法优化）

• 有关sort函数的参数：第三个参数-比较器

• 匿名函数：ctrl+F搜Lambda 函数与表达式

• 还可以使用快速选择nth_element(参考)时间复杂度是O（n）