ACM输入方式小结

输入多组的情况：

import sys

def solve(arr):
    print(arr[0] + arr[1])

for line in sys.stdin:
    arr = line.split()
    arr = list(map(int, arr))
    solve(arr)

已知组数的情况：

import sys

def solve(arr):
    return arr[0] + arr[1]
    
group_nums = input()
group_nums = int(group_nums)
for i in range(group_nums):
    arr = sys.stdin.readline().strip().split(' ')
    arr = list(map(int, arr))
    res = solve(arr)
    print(res)

 遇到特定情况跳出：

import sys

def solve(arr):
    print(arr[0] + arr[1])

group_nums = 0
arr = []
for line in sys.stdin:
    arr = line.split()
    arr = list(map(int, arr))
    if arr != [0, 0]:
        solve(arr)
    else:
        break

import sys

def solve(num, nums):
    ans = 0
    for i in range(num):
        ans += nums[i]
    print(ans)

for line in sys.stdin:
    arr = line.split()
    arr = list(map(int, arr))
    if arr[0] != 0:
        num = arr[0]
        nums = arr[1:num+1]
        solve(num, nums)
    else:
        break
    

字符串处理：

import sys

def solve(arr):
    arr.sort()
    for val in arr:
        print(val, end = ' ')

num = input()
arr = sys.stdin.readline().strip().split(' ')
# strip去掉前后空格和最后的换行符
# split以空格为划分元素的依据
# arr是一个列表
solve(arr)

import sys

def solve(arr):
    arr.sort()
    for val in arr:
        print(val, end = ' ')
    print('') # 这里再添加换行符就空两行了

for line in sys.stdin:
    arr = line.strip().split(' ')
    solve(arr)

import sys

def solve(arr):
    arr.sort()
    num = len(arr)
    for i in range(num):
        if i != num-1:
            print(arr[i], end = ',')
        else:
            print(arr[i])

for line in sys.stdin:
    arr = line.strip().split(',')
    solve(arr)