Cặp số tương thích

Xem dạng PDF

Gửi bài giải

Điểm: 0,15 (OI)
Giới hạn thời gian: 1.0s
Giới hạn bộ nhớ: 256M
Input: stdin
Output: stdout

Nguồn bài:
HSGC2 PhuTho 2023 - 2024
Dạng bài
Ngôn ngữ cho phép
C, C++, Go, Java, Kotlin, Pascal, PyPy, Python, Rust, Scratch

Cho ~n~ số nguyên dương ~a_1, a_2, ..., a_n~ và một số nguyên dương ~k~. Một cặp số ~(a_i, a_j), 1 \le i, j \le n~ được gọi là tương thích với nhau nếu thỏa mãn đồng thời ~2~ điều kiện:

  • ~0 < j - i \le k~;
  • ~a_i = a_j~.

Yêu cầu: Hãy Đếm số lượng cặp số tương thích trong các số đã cho.

Dữ liệu:

  • Dòng đầu chứa hai số nguyên dương ~n~ và ~k~ (~2 \le n \le 3 \times 10^5; 1 \le k \le n~);
  • Dòng thứ hai chứa ~n~ số nguyên dương ~a_1, a_2, ..., a_n~ (~a_i \le 10^5; 1 \le i \le n~).

Kết quả:

  • Ghi một số nguyên duy nhất là kết quả của bài toán.

Ví dụ:

Input
4 2
3 3 3 3
Output
5
Input
6 3
7 5 6 5 7 6  
Output
2
Giải thích:
  • Test ví dụ 1: Có 5 cặp số tương thích là: ~(a_1, a_2)~, ~(a_1, a_3)~, ~(a_2, a_3)~, ~(a_2, a_4)~, ~(a_3, a_4)~.
  • Test ví dụ 2: Có 2 cặp số tương thích là: ~(a_2, a_4)~, ~(a_3, a_6)~.

Ràng buộc:

  • Subtask 1: có 50% số điểm ứng với ~2 \le n \le 5 \times 10^3~;
  • Subtask 2: có 50% số điểm ứng với ~5 \times 10^3 < n \le 3 \times 10^5~.

Bình luận

Hãy đọc nội quy trước khi bình luận.


Không có bình luận tại thời điểm này.