面试题头条最大映射C++ 简洁做法

[编程题] 最大映射
时间限制：1秒

空间限制：32768K

有 n 个字符串，每个字符串都是由 A-J 的大写字符构成。现在你将每个字符映射为一个 0-9 的数字，不同字符映射为不同的数字。这样每个字符串就可以看做一个整数，唯一的要求是这些整数必须是正整数且它们的字符串不能有前导零。现在问你怎样映射字符才能使得这些字符串表示的整数之和最大？

输入描述:
每组测试用例仅包含一组数据，每组数据第一行为一个正整数 n ， 接下来有 n 行，每行一个长度不超过 12 且仅包含大写字母 A-J 的字符串。 n 不大于 50，且至少存在一个字符不是任何字符串的首字母。

输出描述:
输出一个数，表示最大和是多少。

输入例子1:
2
ABC
BCA

输出例子1:
1875

代码 ：

#include <iostream>
#include <String> 
#include <algorithm>
#include <cmath>
using namespace std;
struct weight{
	long long num;
	bool head;
}A[10];
int cmp(const weight &a,const weight &b)
{
    return a.num<b.num;
}
int main() {
	int n;
	cin >> n;
	string str;
	long long MAX = 0;
	while(n--) {
		cin >> str;
		for(int i=str.length()-1; i>=0; i--) {
			A[str[i]-'A'].num += pow(10, str.length()-i-1);
			if(i==0)
				A[str[i]-'A'].head = true;
		}
	}
	sort(A, A+10,cmp);
	if(A[0].head){
		int i;
		for(i=1;i<10;i++){
			if(!A[i].head){
				swap(A[0],A[i]);
				break;
			}
		}
		sort(A+1, A+i+1,cmp);
	}
	for(int i=0; i<=9; i++) {
		MAX += (A[i].num * i);
	} 
	cout << MAX;
}