题目链接:
本题如果用暴力法来做,是
O
(
n
2
)
O(n^2)
O(n2)的,因此需要用双指针优化。
优化后j无法回退。
本题需要注意的是,两组数都是单调递增,如现将指针
i
i
i指向第一组数头部,指针
j
j
j指向第二组数头部,则在
i
=
0
i=0
i=0的第一组循环后,第二组的
j
j
j仍需从
j
=
0
j=0
j=0开始遍历,无法优化。
4 5 6
1 2 4 7
3 4 6 8 9
因此双指针一个 i i i指头,一个 j j j指尾。只要 a [ i ] + b [ j ] > x , a[i]+b[j] > x, a[i]+b[j]>x,就把 j j j左移,否则就把 i i i右移。 ( i (i (i控制增, j j j控制减 ) ) )
还需注意的是while循环里j的取值需要限定在≥0,不然可能会越界
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int a[N], b[N];
int main()
{
int n, m, x;
scanf("%d%d%d", &n, &m, &x);
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
for (int i = 0; i < m; i++) scanf("%d", &b[i]);
for (int i = 0, j = m - 1; i < n; i++) {
while (j >= 0 && a[i] + b[j] > x) j --;
if (a[i] + b[j] == x) printf("%d %d", i, j);
}
return 0;
}
法二:二分
可以对于每一个
i
i
i指向的元素,在第二个数组里面寻找一个数使得
b
[
j
]
=
x
−
a
[
i
]
.
b[j] = x-a[i].
b[j]=x−a[i]. 于是想到二分。
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int a[N], b[N];
int main()
{
int n, m, x;
scanf("%d%d%d", &n, &m, &x);
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
for (int i = 0; i < m; i++) scanf("%d", &b[i]);
for (int i = 0; i < n; i++) {
if (a[i] < x) {
int l = 0, r = m - 1;
while (l < r) {
int mid = l + r >> 1;
if ( a[i] + b[mid] >= x) r = mid;
else l = mid + 1;
}
if (b[l] == x - a[i]) printf("%d %d", i, l);
}
}
return 0;
}