題目描述
“水果消除”是一款手機遊戲,相信大家都玩過或玩過類似的遊戲。
下面是“水果消除”遊戲的一種初始狀態。
消除的基本規則:如果有2個或2個以上的相同水果連在一起,則可以點選並消除。
請問在某一種狀態下,有幾種可以點選並消除的選擇方案。
例如,對於上圖所示的初始狀態,將有6種點選並消除的選擇方案。這6種方案依次如下圖所示。
輸入
先輸入一個整數n,表示放水果的格子總數爲n*n。n取3到1000之間的整數(含3和1000)。
然後依次輸入n*n個表示水果的數據,不同的水果用不同的數字表示,同一種水果用相同的數字表示。
表示水果的數字編號從1開始,不超過100。
輸出
在輸入數據對應的初始狀態下,有幾種點選並消除的選擇方案。
輸出方案數。
樣例輸入
6
1 1 2 2 2 2
1 3 2 1 1 2
2 2 2 2 2 3
3 2 3 3 1 1
2 2 2 2 3 1
2 3 2 3 2 2
樣例輸出
6
提示
#include <iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int a[1003][1003],cnt,flag,n,ans;
int zan[4][2]= {1,0,0,1,-1,0,0,-1};
bool cha(int x,int y,int num)
{
for(int i=0; i<4; i++)
{
if(y+zan[i][1] < 0 || y+zan[i][1] >= n || x+zan[i][0] < 0 || x+zan[i][0] >= n ) continue;
if(a[x+zan[i][0]] [y+zan[i][1]] == num) return true;
}
return false;
}
void dfs(int x,int y,int cnt,int data)
{
if(y<0 || y>=n || x<0 || x>=n)
{
if(cnt > 1) flag=1;
return ;
}
if(a[x][y]==data)
{
cnt++;
a[x][y]=0;
if(!cha(x,y,data))
{
if(cnt > 1) flag=1;return ;
}
for(int i=0; i<4; i++) dfs(x+zan[i][0],y+zan[i][1],cnt,data);
}
}
int main()
{
while(cin >> n)
{
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
cin >>a[i][j];
ans=0;
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
{
if(a[i][j])
{
dfs(i,j,0,a[i][j]);
if(flag) ans++,flag=0;
}
}
printf("%d\n",ans);
}
return 0;
}
下面是我在寫了pOJ NO.2386 Lake counting 之後,改了一下代碼,發現也行的通,邏輯上也清晰了很多
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n,m;
int st[1002][1002];
int dir[4][2]= {{1,0},{-1,0},{0,1},{0,-1}};
int flag;
void dfs(int x,int y,int data,int cnt) {
st[x][y]=0;
cnt++;
for(int i=0; i<4; i++) {
int newx=x+dir[i][0];
int newy=y+dir[i][1];
if(newx>=0 && newx <n && newy>=0 && newy <n && st[newx][newy]==data) {
dfs(newx,newy,data,cnt);
}
}
if(cnt >1) flag=1;
return ;
}
int main() {
while(scanf("%d",&n)!=EOF) {
int ans=0;
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
scanf("%d",&st[i][j]);
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
if(st[i][j]) {
dfs(i,j,st[i][j],0);
if(flag)
{
ans++; flag=0;
}
}
printf("%d\n",ans);
}
}