sort函数妙解数位排序

解题思路:在sort函数中添加优化的比较函数cmp

注意事项:数组要比十万大一点

参考代码:

#include <iostream>
#include<algorithm>
using namespace std; 
int arr[200000];//大数组定义在主函数外
bool cmp(int a,int b)
{
	int a1=a;
	int b1=b;
	int suma=0;//存储a数位和
	int sumb=0;//存储b数位和
	while(a1!=0)
	{
		suma+=a1%10;
		a1=a1/10;
	}
	while(b1!=0)
	{
		sumb+=b1%10;
		b1=b1/10;
	}
	if(suma == sumb)先判断数位和
	return a<b;
	else
	return suma < sumb;
}
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		arr[i]=i;
	}
	sort(arr+1,arr+n+1,cmp);//调用含有比较函数的sort函数
	cout<<arr[m]<<endl;//输出题目要求的目标
	return 0;
}