#CT00007. Khoảng cách cùng nhóm
Khoảng cách cùng nhóm
Khoảng cách cùng nhóm
Phiên bản: Phước Hưng OJ Extended
Đề bài
Có học sinh đứng trên một hàng, học sinh thứ đứng tại vị trí .
Mỗi học sinh thuộc đúng một trong nhóm. Gọi là chỉ số nhóm của học sinh thứ .
Với hai học sinh ở vị trí và thuộc cùng một nhóm, khoảng cách giữa hai học sinh được xác định là:
Hãy tính tổng khoảng cách của tất cả các cặp học sinh thuộc cùng một nhóm.
Input
Dòng đầu tiên chứa hai số nguyên dương và , cách nhau bởi một dấu cách.
Dòng thứ hai chứa số nguyên , cách nhau bởi dấu cách, trong đó là chỉ số nhóm của học sinh thứ .
Output
In ra một số nguyên duy nhất — tổng khoảng cách của tất cả các cặp học sinh thuộc cùng một nhóm:
Subtask
Trong tất cả các Subtask: và .
- Subtask 1 — 1,5 điểm: .
- Subtask 2 — 1,0 điểm: .
- Subtask 3 — 1,5 điểm: Không có ràng buộc bổ sung.
Ví dụ
Input
8 3
1 2 1 1 2 3 1 2
Output
31
Giải thích
Nhóm gồm các học sinh ở vị trí . Tổng khoảng cách giữa các cặp học sinh của nhóm này là:
Nhóm gồm các học sinh ở vị trí . Tổng khoảng cách là:
Nhóm chỉ có một học sinh ở vị trí , nên không tạo ra cặp học sinh nào.
Do đó:
Liên quan
Trong các cuộc thi sau: