紫书 例题8-3 UVa 1152(中途相遇法)
这道题要逆向思维, 就是求出答案的一部分, 然后反过去去寻找答案存不存在。 其实很多其他题都用了这道题目的方法, 自己以前都没有发现, 这道题专门考这个方法。这个方法可以没有一直往下求, 可以省去很多时间。紫书里面把这叫做中途相遇法,双向广搜有点这个方法的味道。这里用到了二分查找, 总的时间复杂度是n的二次方乘logn
#include<cstdio>
#include<vector>
#include<algorithm>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
using namespace std;
const int MAXN = 4123;
int a[5][MAXN], n;
vector<int> g;
inline int find(int key)
{
int l = lower_bound(g.begin(), g.end(), key) - g.begin();
int r = upper_bound(g.begin(), g.end(), key) - g.begin();
return r - l;
}
int main()
{
int T, kase = 0;
scanf("%d", &T);
while(T--)
{
if(kase) puts(""); kase = 1;
scanf("%d", &n);
REP(i, 0, n)
REP(j, 0, 4)
scanf("%d", &a[j][i]);
g.clear();
REP(i, 0, n)
REP(j, 0, n)
g.push_back(a[0][i] + a[1][j]);
int ans = 0;
REP(i, 0, n)
REP(j, 0, n)
ans += find(-a[2][i] - a[3][j]);
printf("%d\n", ans);
}
return 0;
}