紫书 例题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;	
} 

posted @ 2018-04-29 11:42  Sugewud  阅读(118)  评论(0编辑  收藏  举报