結果
| 問題 | 
                            No.1823 Tricolor Dango
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2022-01-29 20:56:46 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 786 bytes | 
| コンパイル時間 | 516 ms | 
| コンパイル使用メモリ | 82,508 KB | 
| 実行使用メモリ | 116,312 KB | 
| 最終ジャッジ日時 | 2025-01-02 16:32:42 | 
| 合計ジャッジ時間 | 77,255 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge5 / judge1 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | WA * 1 | 
| other | TLE * 25 | 
ソースコード
from sys import stdin
import heapq
def inputSynario():
    input = stdin.readline
    num_list = []
    T = int(input().rstrip())
    for i in range(T * 2):
        num_list.append(list(map(int, input().rstrip().split())))
    return num_list
synario = inputSynario()
for synarioId in range(len(synario)//2):
    N=synario[synarioId * 2]
    A=synario[synarioId * 2 + 1]
    isLoopFlag=True
    while(isLoopFlag):
        dango=heapq.nlargest(3,A)
        if(dango[0]==0):
            print("yes")
            isLoopFlag=False
            break
        for i in range(3):
            if(dango[i] != 0):
                index = A.index(dango[i])
                A[index] -= 1
            else:
                print("no")
                isLoopFlag=False
                break