Ban tổ chức Hội thao: Phân loại số báo danh vận động viên

Ban tổ chức Hội thao trường THPT ghi danh sách số báo danh (số nguyên từ 1 đến 1000) của các vận động viên tham gia. Ban tổ chức muốn tính:

• Nếu mode = 0: tổng các số báo danh là số nguyên tố.
• Nếu mode khác 0: tổng các số báo danh không phải số nguyên tố.

Yêu cầu: viết hàm prime(n) kiểm tra số nguyên tố (đã học ở bài trước) và hàm tong_theo_dieu_kien(danh_sach, mode) sử dụng lại hàm prime để tính tổng theo yêu cầu. Chương trình chính chỉ gọi hàm, không viết lại logic kiểm tra nguyên tố hay cộng tổng ngoài hàm.

Định dạng input:

• Dòng 1: số nguyên n - số lượng vận động viên.
• Dòng 2: n số báo danh cách nhau bởi dấu cách.
• Dòng 3: giá trị mode (0 hoặc khác 0).

Định dạng output: in ra một số nguyên là tổng tương ứng.