PAT A1009:Product of Polynomials

Python优雅不优雅?从多项式乘法实现用的迭代器来看

题目内容

This time, you are supposed to find $A\times B$ where $A$ and $B$ are two polynomials.

Input Specification:

Each input file contains one test case. Each case occupies 2 lines, and each line contains the information of a polynomial:

$K$ $N_1 a_{N_1} N_2 a_{N_2} \cdots N_K a_{N_K}$

where $K$ is the number of nonzero terms in the polynomial, $N_i$ and $a_{N_i} (i=1,2,\cdots,K)$ are the exponents and coefficients, respectively. It is given that $1\leqslant K\leqslant 10$,$0\leqslant N_K < \cdots < N_2 < N_1 \leqslant 1000$.

Output Specification:

For each test case you should output the product of $A$ and $B$ in one line, with the same format as the input. Notice that there must be NO extra space at the end of each line. Please be accurate up to 1 decimal place.

Sample Input:

2 1 2.4 0 3.2
2 2 1.5 1 0.5

Sample Output:

3 3 3.6 2 6.0 1 1.6

题目要点

本题 25 分,是比较简单的模拟题。在输入时,Python可以不用考虑给定的$K$,直接用数组切片隔位取指数列表、系数列表,再用zip()构造字典。

重点需要考虑的是如何将多项式的指数和对应系数映射。C/C++可以用数组下标作为指数,数组浮点数值为对应系数。这里Python使用字典比较方便,而且考虑到两个多项式的指数项不一定相同,在求和时要指数项系数对应相加,Python可以使用集合的求并集|运算符实现。

注意,测试点6是非零系数项个数,也就是$K$为0的情况,这时候只输出0。如果有多余空格会出现“格式错误”。

AC 源代码

from itertools import product

ins1 = input()
ins2 = input()
a = zip( map(int, ins1.split()[1::2]), map(float, ins1.split()[2::2]) )
b = zip( map(int, ins2.split()[1::2]), map(float, ins2.split()[2::2]) )

poly = {}
prod = map(lambda x: (x[0][0]+x[1][0], x[0][1]*x[1][1]), product(a, b))

for exp, coef in prod:
    poly[exp] = poly.get(exp, 0.0) + coef

poly = list( filter(lambda x:x[1]!=0.0, sorted(poly.items(), reverse=True)) )
line = ' '.join(map(lambda x: '{:d} {:.1f}'.format(*x), poly))
k = len(poly)

if k:
    print(k, line)
else:
    print(0)

参考来源

A1002多项式加法的解析:https://column.mowchan.me/post/2021/pat-a1002.html