华为笔试题目
幼儿园小朋友占成一列,按位置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