题目链接
这个很棒,知道了用三位数组存储二维图的方法。
用dfs
import java.util.*;
import java.math.*;
import java.io.*;
public class Main {
public static int N,K,R,niu,map[][][]=new int[102][102][4],cow[][]=new int[102][202],is[][];
public static int dx[]={-1,1,0,0};
public static int dy[]={0,0,-1,1};
public static void main(String[] args){
Scanner sc=new Scanner(System.in);
Map<Integer,Integer> m=new HashMap<Integer,Integer>();
N=sc.nextInt();
K=sc.nextInt();
R=sc.nextInt();
is=new int[N+1][N+1];
for(int i=0;i<R;i++){ //0上 1下,2左,3右
int x1,x2,y1,y2;
x1=sc.nextInt();
y1=sc.nextInt();
x2=sc.nextInt();
y2=sc.nextInt();
if(x1==x2){
map[x1][Math.min(y1,y2)][3]=1;
map[x1][Math.max(y1,y2)][2]=1;
}
else{
map[Math.min(x1,x2)][y1][1]=1;
map[Math.max(x1,x2)][y1][0]=1;
}
}
for(int i=0;i<K;i++){
cow[sc.nextInt()][sc.nextInt()]=1;
}
int ji=0;
for(int i=1;i<=N;i++){
for(int j=1;j<=N;j++){
if(is[i][j]==0){
niu=0;
dfs(i,j);
m.put(ji,niu);
ji++;
}
}
}
//System.out.println(m);
int res=0;
for(int i=0;i<m.size();i++){
for(int j=i+1;j<m.size();j++)
res+=m.get(i)*m.get(j);
}
System.out.println(res);
}
public static void dfs(int x,int y){
is[x][y]=1;
if(cow[x][y]==1)
niu++;
for(int i=0;i<4;i++){
if(map[x][y][i]==1)
continue;
int xx=x+dx[i];
int yy=y+dy[i];
if(xx==0||yy==0||xx>N||yy>N||is[xx][yy]==1)
continue;
dfs(xx,yy);
}
}
}