【BZOJ1201】[HNOI2005]数三角形（暴力）

【BZOJ1201】[HNOI2005]数三角形（暴力）

题面

BZOJ
洛谷

题解

预处理每个点向四个方向可以拓展的最大长度，然后钦定一个点作为三角形的某个顶点，暴力枚举三角形长度，检查这样一个三角形是否存在。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define ll long long
#define MAX 1010
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
struct Node{int l,r,d;}p[MAX][MAX];
int n,ans;
int upl[MAX][MAX],upr[MAX][MAX],dnl[MAX][MAX],dnr[MAX][MAX];
int main()
{
	n=read();
	for(int i=1;i<=n;++i)
		for(int j=1;j<=i;++j)
			p[i][j].l=read(),p[i][j].r=read(),p[i][j].d=read();
	for(int i=1;i<=n;++i)
		for(int j=1;j<=i;++j)
			if(p[i][j].l)upl[i][j]=upl[i-1][j]+1;
	for(int i=1;i<=n;++i)
		for(int j=1;j<=i;++j)
			if(p[i][j].r)upr[i][j]=upr[i-1][j-1]+1;
	for(int i=n;i;--i)
		for(int j=1;j<=i;++j)
			if(p[i+1][j].r)dnl[i][j]=dnl[i+1][j+1]+1;
	for(int i=n;i;--i)
		for(int j=1;j<=i;++j)
			if(p[i+1][j+1].l)dnr[i][j]=dnr[i+1][j]+1;
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			for(int k=j;k<=n&&p[i][k].d&&k-j+1<=upl[i][j];++k)
				ans+=(k-j+1)<=upr[i][k];
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			for(int k=j;p[i][k].d&&k<=n&&k-j+1<=dnl[i][j];++k)
				ans+=(k-j+1)<=dnr[i][k];
	printf("%d\n",ans);
	return 0;
}