搜索【洛谷P2845】 [USACO15DEC]Switching on the Lights 开关灯

P2845 [USACO15DEC]Switching on the Lights 开关灯

题目背景

来源：usaco-2015-dec

Farm John 最近新建了一批巨大的牛棚。这些牛棚构成了一个N*N的矩形网络。（1<n<100)

然而bessie十分怕黑，他想计算可以把多少个牛棚的灯打开。

题目描述

有NN个房间，组成了一张NN的网格图，Bessie一开始位于左上角(1,1)，并且只能上下左右行走。

一开始，只有(1,1)这个房间的灯是亮着的，Bessie只能在亮着灯的房间里活动。

有另外M条信息，每条信息包含四个数a,b,c,d，表示房间(a,b)里有房间(c,d)的灯的开关。

请计算出最多有多少个房间的灯可以被打开

水题，但是题意理解错了WA了好长时间。

code:

#include <iostream>
#include <cstdio>
#include <queue>

using namespace std;

const int wx=117;

inline int read(){
	int sum=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
	while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
	return sum*f;
}

int f[wx][wx],mpx[200017],mpy[200017];
int head[200017],vis[wx][wx],had[wx][wx];
int dx[]={0,1,0,-1,0};
int dy[]={0,0,1,0,-1};
int n,m,num,ans=1,tot;

struct e{
	int nxt,to;
}edge[200017];

void add(int from,int to){
	edge[++num].nxt=head[from];
	edge[num].to=to;
	head[from]=num;
}

queue<int > q;

void bfs(){
	vis[1][1]=1; had[1][1]=1; q.push(1); q.push(1);
	while(q.size()){
		int ux=q.front(); q.pop();
		int uy=q.front(); q.pop();
		for(int i=head[f[ux][uy]];i;i=edge[i].nxt){
			int v=edge[i].to;
			int ex=mpx[v]; int ey=mpy[v];
			if(!had[ex][ey]){
				had[ex][ey]=1;ans++;
				if(!vis[ex][ey])if(vis[ex-1][ey]||vis[ex+1][ey]||vis[ex][ey-1]||vis[ex][ey+1])vis[ex][ey]=1,q.push(ex),q.push(ey);
			}

		}
		for(int i=1;i<=4;i++){
			int ex=ux+dx[i]; int ey=uy+dy[i];
			if(ex<1||ey<1||ex>n||ey>n||vis[ex][ey]||!had[ex][ey])continue;
			if(vis[ex-1][ey]||vis[ex+1][ey]||vis[ex][ey-1]||vis[ex][ey+1])vis[ex][ey]=1,q.push(ex),q.push(ey);
		}
	}
	printf("%d\n",ans);
}

int main(){
	n=read(); m=read();
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			f[i][j]=++tot; mpx[tot]=i; mpy[tot]=j;
		}
	}
	for(int i=1;i<=m;i++){
		int x,y,z,c;
		x=read(); y=read(); z=read(); c=read();
		add(f[x][y],f[z][c]);
	}
	bfs();
	return 0;
}