20求助
查看原帖
20求助
387037
MrSWdAxiv楼主2022/9/24 12:39

直角

#include<bits/stdc++.h>
using namespace std;
int n,ans;
char ch[1005][1005];
bool check(int x,int y){
	return (1<=x&&x<=n)&&(1<=y&&y<=n)&&(ch[x][y]=='1');
}
void dfs(int x,int y){
	for(int i=1;i<=n;i++){
		if(check(x+i,y+i)&&check(x-i,y-i)&&check(x+i,y-i)&&check(x-i,y+i)){
			if(ch[x+i][y+i]=='1'&&ch[x-i][y-i]=='1'&&ch[x+i][y-i]=='1'&&ch[x-i][y+i]=='1'){
				ans++;
			}else{
				break;
			}
		}
	}
}
void dfs2(int x,int y){
	if(check(x,y+1)&&check(x+1,y)&&check(x+1,y+1)){
		if(ch[x][y+1]=='1'&&ch[x+1][y]=='1'&&ch[x+1][y+1]=='1'){
			ans++;
			for(int i=1;i<=n;i++){
				if(check(x-i,y-i)&&check(x+i+1,y+i+1)&&check(x-i,y+i+1)&&check(x+i+1,y-i)){
					if(ch[x-i][y-i]=='1'&&ch[x+i+1][y+i+1]=='1'&&ch[x-i][y+i+1]=='1'&&ch[x+i+1][y-i]=='1'){
						ans++;
					}else{
						break;
					}
				}
				
			}
		}
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>ch[i][j];
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(ch[i][j]=='1'){
				dfs(i,j);
				dfs2(i,j);
			}
		}
	}
	cout<<ans<<'\n';
	return 0;
}

不要管这凌乱的码风

2022/9/24 12:39
加载中...