PAT (Advanced Level) Practice 1009 Product of Polynomials（Python）

最新推荐文章于 2023-09-05 23:15:01 发布

原创最新推荐文章于 2023-09-05 23:15:01 发布 · 220 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

Python 同时被 2 个专栏收录

8 篇文章

订阅专栏

PAT(Advance)

7 篇文章

订阅专栏

本文详细介绍了一种解决多项式相乘问题的高效算法。通过使用列表存储和处理多项式的系数与指数，文章提供了一个清晰的实现过程，包括输入解析、乘积计算以及输出格式化。示例代码展示了如何利用Python的numpy库进行操作。

程序员入门水平，贴出代码大家一起进步！
原题传送门

题目

This time, you are supposed to find A×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_{N1}\;N_2\;a_{N2}\dots\;N_k\;a_{Nk}$
where K is the number of nonzero terms in the polynomial, $N_i$ and $a_{Ni}\;(i=1,2,⋯,K)$ are the exponents and coefficients, respectively. It is given that $1≤K≤10, 0≤N_K <⋯<N_2<N_1 ≤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

思路||总结

设置20001初始值为0的列表，对于两个因子，将其系数的乘积加在列表位置为两个因子的幂的和的位置。
count一下又多少个非零项，然乎输出。

AC代码

import numpy as np


def main():
    str_0 = input().strip().split()
    str_0[0] = int(str_0[0])
    for i in range(str_0[0]):
        str_0[i * 2 + 1] = int(str_0[i * 2 + 1])
        str_0[i * 2 + 2] = float(str_0[i * 2 + 2])
    str_1 = input().strip().split()
    str_1[0] = int(str_1[0])
    for i in range(str_1[0]):
        str_1[i * 2 + 1] = int(str_1[i * 2 + 1])
        str_1[i * 2 + 2] = float(str_1[i * 2 + 2])
    result = np.zeros(2001)
    count = 0
    for i in range(str_0[0]):
        for j in range(str_1[0]):
            # 对于两个因子，将其系数的乘积加在列表位置为两个因子的幂的和的位置
            result[str_0[i * 2 + 1] + str_1[j * 2 + 1]] += str_0[i * 2 + 2] * str_1[j * 2 + 2]
    # 数一下有多少个非0项
    for i in range(2001):
        if result[i] != 0:
            count += 1
    print(count, end='')
    for i in range(2001):
        if result[2000 - i] != 0:
            print(' %d %.1f' % (2000 - i, result[2000 - i]), end='')


main()