2019 CCPC 秦皇岛 A. Angle Beats

题面链接

题目大意: 在二维平面上 给你n个点 有q次查询 每次查询会有一个坐标Q,询问 有多少点对 和 Q点 能组成直角三角形;

思路:分两个情况:

一:Q点是直角顶点 二:Q点不是直角顶点;

第一个情况:只需要看有多少个和Q点组成的斜率互乘为-1,

第二个情况:就是看Q点和任意一个点 组成的直线的斜率 和 这个点 和 点集合中的点组成的直线有没有存在互乘为-1的

注意点: 会有卡时间需要注意不要用map 建议用哈希查找表 或者 hash_map 或者 unordered_map维护 

时间上(哈希查找表 < hash_map < unordered_map)

/*
Andong 2020/10/27
*/
#include<iostream>
#include<hash_map>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
using namespace __gnu_cxx;
typedef long long ll;
typedef pair<int, int> pii;
const int maxn = 2e3 + 5;
const int mod = 2333;
const ll base = 2e9 + 7;
struct Hash {
	struct Has {
		int num, nxt;
		ll h;
	} has[maxn];
	int head[mod], cnt;
	int rub[maxn], top;
	void init() {

		while(top) head[rub[top--]] = 0;
		cnt = 0;
	}
	void add(ll h) {

		int u = h % mod;
		u = u < 0 ? -u : u;
		for(int i = head[u]; i; i = has[i].nxt) {

			if(has[i].h == h) {
				++has[i].num;
				return;
			}
		}
		rub[++top] = u;
		has[++cnt] = { 1, head[u], h }, head[u] = cnt;
	}
	int get(ll h) {

		int u = h % mod;
		u = u < 0 ? -u : u;
		for(int i = head[u]; i; i = has[i].nxt) {

			if(has[i].h == h) return has[i].num;
		}
		return 0;
	}
} hi[maxn];

int xi[maxn], yi[maxn];
pii pi[maxn];
int n, q;

int gcd(int a, int b) {

	return b ? gcd(b, a % b) : a;
}

pii cal(int a, int b) {

	if(a == 0) b = 1;
	else if(b == 0) a = 1;
	else {

		int d = gcd(abs(a), abs(b));
		a /= d, b /= d;
		if(b < 0) a = -a, b = -b;
	}
	return {a, b};
}

int main() {
	scanf("%d%d", &n, &q);

	for(int i = 1; i <= n; ++i) scanf("%d%d", &xi[i], &yi[i]);
	for(int i = 1; i <= n; ++i) {

		hi[i].init();
		for(int j = 1; j <= n; ++j) {

			if(i == j) continue;
			pii tmp = cal(xi[i] - xi[j], yi[i] - yi[j]);
			hi[i].add(tmp.first * base + tmp.second);
		}
	}
	int p = n + 1;
	while(q--) {

		int x, y;
		scanf("%d%d", &x, &y);
		hi[p].init();
		int ret1 = 0, ret2 = 0;
		for(int i = 1; i <= n; ++i) {

			pi[i] = cal(x - xi[i], y - yi[i]);
			hi[p].add(pi[i].first * base + pi[i].second);
		}
		for(int i = 1; i <= n; ++i) {

			pii tmp = pi[i];
			if(tmp.first == 0 && tmp.second == 1 || tmp.first == 1 && tmp.second == 0);
			else {

				if(tmp.first > 0) tmp.second = -tmp.second;
				else tmp.first = -tmp.first;
			}
			ret1 += hi[i].get(tmp.second * base + tmp.first);
			ret2 += hi[p].get(tmp.second * base + tmp.first);
		}
		printf("%d\n", ret1 + ret2 / 2);
	}

	return 0;
}