PYTHON蓝桥杯——每日一练（简单题）

题目

复数求和

从键盘读入n个复数（实部和虚部都为整数）用链表存储，遍历链表求出n个复数的和并输出。

输入格式

复数的个数和各个复数的值

输出格式

输出实部和和虚部和

代码

n = int(input())
data = [list(map(int,input().split())) for _ in range(n)] #生成一个二维数组
for i in range(n):
    if len(data[i]) ==1:
        data[i].append(0)
x = 0
y = 0
for j in range(n):
    x+= data[j][0] #实部相加
    y+= data[j][1] #虚部相加
print(f'{x}+{y}i')

运行结果