华为笔试题目

华为笔试题目

幼儿园小朋友占成一列，按位置1、2、3顺序编号，找出3位小朋友，拿着相同颜色的糖果，且总数不少于其他任何三位小朋友（相同颜色）的糖果总数，糖果颜色只有2种。
输入：
第一行位小朋友的总人数N，后面的N行分别为第1~N个小朋友拿的糖果数量和糖果颜色
输出：
第一行为3个小朋友位置编号，第二行为糖果颜色，第三行为糖果总数
如没有满足条件的小朋友，输出字符串“null”
python代码：

import numpy as np
#import get_index
n = int(input())
sugnum = list()
sugnum1 = list()
sugnum2 = list()
sugcol = list()
index_1 = list()
renum = list()
for i in range(n):
    x,y = map(int, input().split())
    sugnum.append(x)
    sugcol.append(y)
for i in range(n):
    if sugcol[i] == 1:
        sugnum1.append(sugnum