Tàu điện ngầm Bertown
Đề bài
Mô tả
Hệ thống tàu điện ngầm của thành phố gồm nhà ga, được xây dựng theo quy tắc sau:
- Từ mỗi nhà ga có đúng một chuyến tàu khởi hành, và điểm đến của chuyến tàu đó là nhà ga (có thể ).
- Với mỗi nhà ga , tồn tại đúng một nhà ga sao cho .
Độ tiện lợi của hệ thống là số cặp có thứ tự sao cho một hành khách xuất phát từ nhà ga có thể đến được nhà ga sau khi đi một số chuyến tàu (có thể là chuyến), với .
Trước chuyến thăm của tổng thống, thị trưởng có thể sửa lại tuyến đường bằng cách thay đổi giá trị của không quá hai nhà ga. Sau khi thay đổi, hệ thống vẫn phải thoả mãn cả hai quy tắc trên.
Hãy tính độ tiện lợi lớn nhất mà thị trưởng có thể đạt được.
Dữ liệu vào
- Dòng đầu chứa số nguyên là số nhà ga.
- Dòng thứ hai chứa số nguyên mô tả cấu trúc hiện tại của hệ thống.
Dữ liệu ra
In ra một số nguyên duy nhất là độ tiện lợi lớn nhất có thể đạt được.
Ràng buộc
- , các giá trị đôi một khác nhau.
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 3 2 1 3 |
9 | Ban đầu có hai nhóm nhà ga liên thông là và , cho cặp. Thị trưởng đổi và , gộp tất cả thành một nhóm nên mọi cặp trong cặp đều hợp lệ. |
| 5 1 5 4 3 2 |
17 | Ban đầu có ba nhóm , , , cho cặp. Đổi và để gộp hai nhóm kích thước thành nhóm kích thước , thu được . |
Bình luận