Chỉ số logit lớn nhất (argmax)

Mô tả bài toán

argmax là bước cuối của mọi bài phân loại: sau khi mô hình cho ra vector logits, ta lấy lớp có điểm cao nhất. Khi có nhiều logit bằng nhau, quy ước chọn lớp có chỉ số nhỏ nhất.

Yêu cầu

Đọc từ stdin: dòng 1 là N; dòng 2 gồm N logit. In ra chỉ số (0-based) của logit lớn nhất. Nếu nhiều logit cùng lớn nhất, in chỉ số nhỏ nhất.

Quy ước nộp bài

• Chỉ cần viết một chương trình đọc stdin và in ra stdout. Bài này không yêu cầu viết hàm.
• Không truyền lời nhắc vào input(). Lời nhắc sẽ bị in vào stdout và làm bài sai.
• Chỉ in đúng nội dung ở mục Output. Không in thêm nhãn, dòng trống hay ký tự thừa.
• Output được so khớp từng ký tự, phân biệt chữ hoa/thường và dấu câu.

Ràng buộc

• 1 ≤ N ≤ 10⁴; logit là số thực trong [-10⁴, 10⁴].