【ccf認證】201512-2 消除類遊戲（c/c++）

試題編號：	201512-2
試題名稱：	消除類遊戲
時間限制：	1.0s
內存限制：	256.0MB
問題描述：	問題描述 　　消除類遊戲是深受大衆歡迎的一種遊戲，遊戲在一個包含有n行m列的遊戲棋盤上進行，棋盤的每一行每一列的方格上放着一個有顏色的棋子，當一行或一列上有連續三個或更多的相同顏色的棋子時，這些棋子都被消除。當有多處可以被消除時，這些地方的棋子將同時被消除。 　　現在給你一個n行m列的棋盤，棋盤中的每一個方格上有一個棋子，請給出經過一次消除後的棋盤。 　　請注意：一個棋子可能在某一行和某一列同時被消除。 輸入格式 　　輸入的第一行包含兩個整數n, m，用空格分隔，分別表示棋盤的行數和列數。 　　接下來n行，每行m個整數，用空格分隔，分別表示每一個方格中的棋子的顏色。顏色使用1至9編號。 輸出格式 　　輸出n行，每行m個整數，相鄰的整數之間使用一個空格分隔，表示經過一次消除後的棋盤。如果一個方格中的棋子被消除，則對應的方格輸出0，否則輸出棋子的顏色編號。 樣例輸入 4 5 2 2 3 1 2 3 4 5 1 4 2 3 2 1 3 2 2 2 4 4 樣例輸出 2 2 3 0 2 3 4 5 0 4 2 3 2 0 3 0 0 0 4 4 樣例說明 　　棋盤中第4列的1和第4行的2可以被消除，其他的方格中的棋子均保留。 樣例輸入 4 5 2 2 3 1 2 3 1 1 1 1 2 3 2 1 3 2 2 3 3 3 樣例輸出 2 2 3 0 2 3 0 0 0 0 2 3 2 0 3 2 2 0 0 0 樣例說明 　　棋盤中所有的1以及最後一行的3可以被同時消除，其他的方格中的棋子均保留。 評測用例規模與約定 　　所有的評測用例滿足：1 ≤ n, m ≤ 30。

 

基本思路：

1、考慮到“一個棋子可能在某一行和某一列同時被消除”，故不能邊數邊消。應該選擇把要消除的位置全部記錄下來，然後再一次性消除。

2、定義2個二維數組a、a1，分別記錄棋盤上的棋子和應該消除的位置。初始a1全置爲0，要消除的位置值改爲1。

3、分別對行和列進行掃描，用_count記錄連續相同的棋子個數。如果某行（列）某個棋子處_count>=3，那就把該位置以及它前面的2個位置都在a1中置爲1。換行（列）後_count重置爲1。

4、遍歷棋盤，對a1中值爲1的位置，把a中對應的位置改爲0。

 

AC代碼：

#include <iostream>
#include <cstdio>

using namespace std;

int a[40][40],a1[40][40];//a1記錄要消除的位置
int n,m;

void init(){
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            a[i][j]=0;
            a1[i][j]=0;
        }
    }
}

void read(){
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            scanf("%d",&a[i][j]);

}

void pos(){
    int _count=1,ans=0;
    //橫掃
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(ans==a[i][j]) _count++;
            else{
                ans=a[i][j];
                _count=1;
            }
            if(_count>=3){
                for(int k=j;k>=j-2;k--)
                    a1[i][k]=1;
            }
//            printf("ans=%d, count=%d\n",ans,_count);
        }
        _count=1;
        ans=0;
    }
    //豎掃
    for(int j=1;j<=m;j++){
        for(int i=1;i<=n;i++){
            if(ans==a[i][j]) _count++;
            else{
                ans=a[i][j];
                _count=1;
            }
            if(_count>=3){
                for(int k=i;k>=i-2;k--)
                    a1[k][j]=1;
            }
//            printf("ans=%d, count=%d\n",ans,_count);
        }
        _count=1;
        ans=0;
    }
}

void solve(){
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            if(a1[i][j]==1) a[i][j]=0;
}

void display(){
//    printf("-------------\n");
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            printf("%d ",a[i][j]);
        }
        printf("\n");
    }
//    printf("-------------\n");
//    for(int i=1;i<=n;i++){
//        for(int j=1;j<=m;j++){
//            printf("%d ",a1[i][j]);
//        }
//        printf("\n");
//    }
}

int main(){

    while(scanf("%d%d",&n,&m)!=EOF){
        init();
        read();
        pos();
//        display();
        solve();
        display();
    }

    return 0;
}