本题源自AcWing算法基础课,题号是880。
题目描述:
解法一、双指针
时间复杂度: O(n+m)
单调性:ai + bi = x。总和是一定的,ai增加的时候,要满足条件bi应当是减小。
一个增,一个减。满足单调性,可以用双指针。
AC代码:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int a[N], b[N];
int main()
{
int n, m, x;
cin>> n >> m >> x;
for (int i = 0; i < n; i ++ )
cin>> a[i];
for (int i = 0; i < m; i ++ )
cin>> b[i];
for (int i = 0, j = m-1; i < n; i ++ )
{
while(j >= 0 && a[i] + b[j] > x) j--;
if(j >= 0 && a[i] + b[j] == x) // j >= 0可不写
cout<< i << " " << j << endl;
}
return 0;
}
解法二、STL二分
既然要找两个和为x的数,那么我们可以转换成确定了一个数y的值为 ai,然后快速在 b中找出 x−y的问题。显然我们可以二分查找,算法复杂度为 O(nlogn),可以AC。
AC代码:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int a[N], b[N];
int main()
{
int n, m, x;
cin>> n >> m >> x;
for (int i = 0; i < n; i++)
cin>> a[i];
for (int i = 0; i < m; i++)
cin>> b[i];
for (int i = 0; a[i] < x; i++)
{
int k = lower_bound(b, b+m, x-a[i]) - b;
if(a[i] + b[k] == x)
{
cout<< i << " " << k << endl;
return 0;
}
}
return 0;
}
解法三、哈希
我们可以在输入a时把每个数存进哈希表里。
对于每个输入的 bi,看看 x−bi是否出现在哈希表即可。
AC代码:
#include <iostream>
#include <cstring>
#include <algorithm>
#include <unordered_map>
using namespace std;
const int N = 100010;
int n, m, x;
int a[N], b[N];
unordered_map<int, int> h;
int main() {
cin >> n >> m >> x;
for (int i = 0; i < n; i ++)
{
cin >> a[i];
h[a[i]] = i;
}
for (int i = 0; i < m; i ++)
{
cin >> b[i];
if (h.count(x - b[i]))
cout << h[x - b[i]] << " " << i << endl;
}
return 0;
}
祝我们天天快乐AC~