2023.3.29 牛客小白月赛 69 补题

等腰三角形(hard)

定理一：格点上可以出现的正多边形只有正四边形

根据这个定理我们不需要去判断格点上的等边三角形

思路:

n 最大为 3000 ， 这提示我们要优化复杂度到 $O(n^2)$ , 那么如何优化呢 ， 优化思路就是枚举等腰三角形的顶点(两个相等腰之间的) ， 然后求与其他点之间的距离，每求出一个顶点与其他点之间的距离 ，就可以加上前面求出的相同距离的边数作为贡献 ， 为了减去相同的边 ， 还要给所有的边打标记 ， 根据当前点的中点求出对称点的位置 ， 看这个点是否出现过 ，出现过减去贡献即可；

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
typedef long long ll;
const int N = 2e6+10;
const int M = 3e3 + 10; 
const int mod = 1e9 + 7;
typedef pair<int,int>PII;
const int inf = 1 << 31 - 1;
const double eps = 1e-9;

int n;
int dis(int x,int y,int m,int n){return (x - m) * (x - m) + (y - n) * (y - n);}
int cnt[N] , x[M] , y[M];
bool vis[M][M];
ll ans = 0;

int main(){

	IOS
	
	cin >> n;
	for(int i=1;i<=n;i++){
		cin >> x[i] >> y[i];
	}
	
	ll ans = 0 ; 
	
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(i == j) continue;
			ans += cnt[dis(x[i] , y[i] , x[j] , y[j])];//累加贡献
			cnt[dis(x[i] , y[i] , x[j] , y[j])]++;//计数
			vis[x[j] + 1500][y[j] + 1500] = 1;//打标记
			int xx = x[i] * 2 - x[j] + 1500;
			int yy = y[i] * 2 - y[j] + 1500;
			if(vis[xx][yy]) ans--;//减去重复点
		}
		for(int j=1;j<=n;j++){//还原
			if(i == j) continue;
			vis[x[j] + 1500][y[j] + 1500] = 0;
			cnt[dis(x[i] , y[i] , x[j] , y[j])]--;
		}
	}
	
	cout << ans ;


	
	return 0;
}
//freopen("文件名.in","r",stdin);
//freopen("文件名.out","w",stdout);