給定 n 個整數和目標值 k,求有多少對不同的數 (x, y) 滿足 x + y = k。數對視為無序,即 (x,y) 和 (y,x) 算同一對;x 和 y 可以相等(需至少出現兩次)。輸出滿足條件的不同數對個數。多測試用例。
去重 + 查找:
unique 去重,得到不重複值數組 a[0..cnt-1]。a[i],若 a[i] ≤ k,二分查找 k - a[i] 是否存在於 a 中。或雙指針:i=0, j=cnt-1,根據 a[i]+a[j] 與 k 比較移動指針,O(cnt)。
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
int bin_search(int* a, int l, int r, int key) {
while (l <= r) {
int mid = (l + r) >> 1;
if (a[mid] == key) return mid;
else if (a[mid] > key) r = mid - 1;
else l = mid + 1;
}
return -1;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int T; cin >> T;
while (T--) {
int n, k; cin >> n >> k;
vector a(n);
for (int i = 0; i < n; ++i) cin >> a[i];
sort(a.begin(), a.end());
a.erase(unique(a.begin(), a.end()), a.end());
int cnt = 0;
for (int x : a) {
if (x <= k && bin_search(a.data(), 0, a.size()-1, k - x) != -1)
++cnt;
}
cout << cnt << '\n';
}
return 0;
}
輸入:4 4 / 2 2 2 2
去重後 [2],2+2=4,存在 → 輸出 1。正確。
排序 O(n log n),去重 O(n),查找 O(cnt log cnt) 或 O(cnt)。n ≤ 100000 完全滿足。
x 和 y 可相等,但原數組需至少有兩個該值(去重後邏輯自動處理:x == k-x 時只要集合中存在即可,因為去重不影響存在性)。