PAT乙级 1056 组合数的和 (15 分) 巧妙解法 Python / C

1056 组合数的和 (15 分)

给定 N 个非 0 的个位数字，用其中任意 2 个数字都可以组合成 1 个 2 位的数字。要求所有可能组合出来的 2 位数字的和。例如给定 2、5、8，则可以组合出：25、28、52、58、82、85，它们的和为330。

输入格式：

输入在一行中先给出 N（1 < N < 10），随后给出 N 个不同的非 0 个位数字。数字间以空格分隔。

输出格式：

输出所有可能组合出来的2位数字的和。

输入样例：

3 2 8 5

输出样例：

330

作者: CHEN, Yue

单位: 浙江大学

时间限制: 400 ms

内存限制: 64 MB

代码长度限制: 16 KB

---

分析：

    这N个不重复的个位数，与其他数组成两位数。每个数字有且只有2*(N-1)种，其中作为十位数有(N-1)种；作为个位数也是(N-1)种。
    所以每个数字(设为a,0<a<10)对总和的贡献为：作为十位数是的 10*a*(N-1) 加上作为个位数的 a*(N-1)，即对总和的贡献为 11*(N-1)*a*
    所以只需要累加所有a，再乘上11*(N-1)即为最终结果。
 

---

Python3代码：

A=[int(a) for a in input().split()]	#读入一行整数（包括第一个的N）
sum=0
for a in A[1:]: sum+=a	#累加所有a(注意不包含N)
print(sum*11*(A[0]-1))	#再乘上11*(N-1)

对Python3更熟练后，我又“精简”了一下，只有两行代码就能AC，不过可读性较差，理解还是按上面来较合适。

A=[int(a) for a in input().split()]	#读入一行整数（包括第一个的N）
print(sum(A[1:])*11*(A[0]-1))	#累加所有a(注意不包含N)*11*(N-1)

C语言代码（兼容C++）：

考虑到很多学生没学Python，还是写了个C语言版本，顺便练一练（不用IDE直接AC）。

#include<stdio.h>
int main(){
    int N,sum=0;    //sum 即为N个数的和
    scanf("%d",&N);
    for(int i=0;i<N;i++){
        int a;  scanf("%d",&a);
        sum +=a;    //累加所有a
    }
    printf("%d",sum*11*(N-1));
    return 0;
}