当前位置: 代码迷 >> 综合 >> 紫书 例题8-3 UVa 1152(中途相遇法)
  详细解决方案

紫书 例题8-3 UVa 1152(中途相遇法)

热度:10   发布时间:2023-09-20 22:21:06.0

这道题要逆向思维, 就是求出答案的一部分, 然后反过去去寻找答案存不存在。 其实很多其他题都用了这道题目的方法, 自己以前都没有发现, 这道题专门考这个方法。这个方法可以没有一直往下求, 可以省去很多时间。紫书里面把这叫做中途相遇法,双向广搜有点这个方法的味道。这里用到了二分查找, 总的时间复杂度是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;	
}