Một trai giam tội phạm có lệnh tiếp nhân 50 tội phạm, trong đó có ba loại là trộm cắp, ma túy, phạm luật giao thông. Để tổ chức giam giữ và quản lý thuận lợi, trưởng trại giam cần chia 50 tội phạm thành 3 nhóm riêng biệt để giam vào 3 khu vực. Em hãy viết chương trình giúp trưởng trại giam thực hiện công việc trên