牛客网输入输出问题（python版）

牛客网输入输出问题（python版）

牛客网没有模板，这里记录一种方法。

关于输入形式为“第一行为数据组数，接下来几行为数据”的题目。

用个例题：需要买n朵花，且分k（1 输入：多组输入。第一行一个正整数T（1<=T<=10^5)，表示数据组数。
接下来T行，每行一个正整数n（1<=n<=10^9)，表示预计买花的数量。
输出：每组数据输出一行，共T行。
判断能否刚好买到n朵花，可以则输出"YES"，否则输出"NO"。

示例：
输入：2 （数据组数T）
20 （花的数量n）
21 （花的数量n）
输出：YES
NO

l1=[int(i) for i in input().split()]
T=l1[0]
print(T)
for m in range(T):
    n = [int(i) for i in input().split()][0]
    print(n)

• 用input().split()获取输入
  （这里将输入的数据作为列表，仅演示输入）