Tập xe
Xem PDF
Điểm:
300 (p)
Thời gian:
1.0s
Bộ nhớ:
256M
Input:
bàn phím
Output:
màn hình
Cô giáo trường tiểu học \(X\) đang dạy \(n\) học sinh tập xe đạp, các học sinh được đánh số từ \(1\) tới \(n\), học sinh thứ \(𝑗\) có trọng lượng là \(a_𝑗\). Có một xe đạp duy nhất với tải trọng là \(m\), hai học sinh chỉ có thể cùng lên xe nếu tổng trọng lượng của hai học sinh không vượt quá \(m\).
Cô giáo tự hỏi có bao nhiêu cách chọn hai học sinh khác nhau cho cùng lên xe, sau nhiều giờ tính toán không có kết quả, cô quyết định hỏi các chuyên gia lập trình giải bài toán Counting Student Pairs (CSP)
Yêu cầu: Đếm số cặp chỉ số \(i, j\) trong đó \(i < j\) và \(a_i + a_j \leq m\)
Input
- Dòng 1 chứa hai số nguyên dương \(n, m\ (m \leq 10^6)\)
- Dòng 2 chứa \(n\) số nguyên dương \(a_1, a_2, \ldots, a_n\) (\(\forall{i}: a_i \leq 10^6\))
Output
Ghi một số nguyên duy nhất là đáp số
Scoring
- Subtask #1 (\(60\%\) số điểm): \(n \leq 10^4\).
- Subtask #2 (\(20\%\) số điểm): \(n \leq 10^5\).
- Subtask #3 (\(20\%\) số điểm): \(n \leq 10^6\).
Example
Test 1
Input
5 6
1 2 3 4 5Output
6
Bình luận
Mình xin chia sẻ lời giải bài này như sau:
Đầu tiên, ta có nhận xét rằng, việc sắp xếp lại thứ tự các phần tử của mảng theo thứ tự tăng dần không làm ảnh hưởng đến kết quả của bài toán !
Tiếp theo, ý tưởng của bài này là sử dụng tìm kiếm nhị phân như sau:
Gọi \(P(j,i)\) là số phần tử \(a_j\) thoả mãn \(a_j\le m-a_i\) và \(Q(j,i)\) là số phần tử \(a_j\) thoả mãn \(a_j>m-a_i\)
Khi đó ta có: \(P(j,i)+Q(j,i)=n-i+1 (\text{ với }0\le i<j<n)\).
Và để tính được \(Q(j,i)\) ta sử dụng chặt nhị phân để tìm. Và sau khi tính được \(Q(j,i)\) ta dễ dàng suy ra được \(P(j,i)\)
Và kết quả của bài toán chính là: \(\sum\limits_{0\le i<j<n}P(j,i)\)
Độ phức tạp của bài toán là: \(O(nlog(n))\)
Các bạn có thể tham khảo code tại đây: Link
Chú ý: Ở trong code của mình có sử dụng hàm: upper_bound, chính là hàm tìm kiếm nhị phân, các bạn có thể tự tìm hiểu thêm về những hàm tương tự như: lower_bound,...
Ps: Nếu có gì thắc mắc, các bạn cứ comment nhé
-
9
VoBaThongL921
9:50 a.m. 30 Tháng 8, 2021
13 bình luận nữaCảm ơn anh nhiều nhé. Nhờ code của anh mà em học được nhiều thứ lắm