【LOJ#3266】「USACO 2020.2 Platinum」Equilateral Triangles（结论，枚举，模拟）

最新推荐文章于 2022-07-24 08:55:16 发布

DD(XYX)

最新推荐文章于 2022-07-24 08:55:16 发布

阅读量212

点赞数 1

分类专栏：模拟文章标签：算法 c++ USACO

本文链接：https://blog.csdn.net/weixin_43960414/article/details/125712035

版权

模拟专栏收录该内容

31 篇文章 1 订阅

订阅专栏

题面

🔗

题解

我们大致有一个 $n^4$ 的做法。

枚举两个点，知道了二者的距离 $d$ ，然后，我们知道与一个点距离为 $d$ 的点围成了一个正方形，那么两个正方形的交点就是第三个点可能存在的位置，如下图：

除了意识到答案是 $O(n^4)$ 级别然后下意识开了 $\rm long~long$ 以外，我们还可以发现：

一个“等边”三角形内必定存在两个点要么 $x - y$ 相等要么 $x + y$ 相等，也就是处于同一条45°斜线。根据上图分类讨论易证。

所以，以 $x - y$ 相等为例，我们可以 $O(n^3)$ 枚举这样的两个点，然后统计绿线上的点的数量：

我们差分一下：
在这里插入图片描述
本质不同的蓝线只有 $O(n^2)$ 个，我们可以预处理每条蓝线上的点的数量，预处理复杂度 $O(n^2)$ 。总时间复杂度 $O(n^3)$ 。

别忘了把 $x + y$ 相等的两个点也枚举一下，再去个重。

CODE

#include<map>
#include<set>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<random>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<unordered_map>
// #pragma GCC optimize(2)
using namespace std;
#define MAXN 305
#define LL long long
#define ULL unsigned long long
#define ENDL putchar('\n')
#define DB double
#define lowbit(x) (-(x) & (x))
#define FI first
#define SE second
#define PR pair<int,int>
#define UIN unsigned int
int xchar() {
	static const int maxn = 1000000;
	static char b[maxn];
	static int pos = 0,len = 0;
	if(pos == len) pos = 0,len = fread(b,1,maxn,stdin);
	if(pos == len) return -1;
	return b[pos ++];
}
// #define getchar() xchar()
inline LL read() {
	LL f = 1,x = 0;int s = getchar();
	while(s < '0' || s > '9') {if(s<0)return -1;if(s=='-')f=-f;s = getchar();}
	while(s >= '0' && s <= '9') {x = (x<<1) + (x<<3) + (s^48);s = getchar();}
	return f*x;
}
void putpos(LL x) {if(!x)return ;putpos(x/10);putchar((x%10)^48);}
inline void putnum(LL x) {
	if(!x) {putchar('0');return ;}
	if(x<0) putchar('-'),x = -x;
	return putpos(x);
}
inline void AIput(LL x,int c) {putnum(x);putchar(c);}

int n,m,s,o,k;
char a[305][305];
int vc[605][305],cn[605];
int l[605][605],r[605][605];
int main() {
	n = read();
	for(int i = 1;i <= n;i ++) {
		scanf("%s",a[i] + 1);
		for(int j = 1;j <= n;j ++) {
			if(a[i][j] == '*') {
				vc[i-j+n][++ cn[i-j+n]] = i+j;
			}
		}
	}
	for(int i = 1;i <= n+n;i ++) {
		vc[i][++ cn[i]] = n+n+1;
		for(int j = 1;j <= cn[i];j ++) {
			int ll = (j==1 ? 1:vc[i][j-1]),rr = vc[i][j];
			l[i][rr] = j; r[i][ll] = j-1;
			for(int k = ll+1;k < rr;k ++) {
				l[i][k] = j; r[i][k] = j-1;
			}
		}
	}
	LL ans = 0;
	for(int i = 1;i <= n;i ++) {
		for(int j = 1;j <= n;j ++) {
			if(a[i][j] == '*') {
				int x = i+1,y = j+1,d = 2,ii = i-j+n;
				while(x <= n && y <= n) {
					if(a[x][y] == '*') {
						int ll,rr;
						if(ii > d) {
							ll = l[ii-d][i+j],rr = r[ii-d][x+y];
							if(ll <= rr) ans += rr-ll+1;
						}
						if(ii + d <= n+n) {
							ll = l[ii+d][i+j],rr = r[ii+d][x+y];
							if(ll <= rr) ans += rr-ll+1;
						}
					}
					x ++; y ++; d += 2;
				}
			}
		}
	}
	for(int i = 1;i <= n+n;i ++) cn[i] = 0;
	for(int i = 1;i <= n;i ++) {
		for(int j = 1;j+j <= n;j ++) swap(a[i][j],a[i][n-j+1]);
		for(int j = 1;j <= n;j ++) {
			if(a[i][j] == '*') {
				vc[i-j+n][++ cn[i-j+n]] = i+j;
			}
		}
	}
	for(int i = 1;i <= n+n;i ++) {
		vc[i][++ cn[i]] = n+n+1;
		for(int j = 1;j <= cn[i];j ++) {
			int ll = (j==1 ? 1:vc[i][j-1]),rr = vc[i][j];
			l[i][rr] = j; r[i][ll] = j-1;
			for(int k = ll+1;k < rr;k ++) {
				l[i][k] = j; r[i][k] = j-1;
			}
		}
	}
	for(int i = 1;i <= n;i ++) {
		for(int j = 1;j <= n;j ++) {
			if(a[i][j] == '*') {
				int x = i+1,y = j+1,d = 2,ii = i-j+n;
				while(x <= n && y <= n) {
					if(a[x][y] == '*') {
						int ll,rr;
						if(ii > d) {
							ll = l[ii-d][i+j+1],rr = r[ii-d][x+y-1];
							if(ll <= rr) ans += rr-ll+1;
						}
						if(ii + d <= n+n) {
							ll = l[ii+d][i+j+1],rr = r[ii+d][x+y-1];
							if(ll <= rr) ans += rr-ll+1;
						}
					}
					x ++; y ++; d += 2;
				}
			}
		}
	}
	AIput(ans,'\n');
	return 0;
}