HUT-XXXX 俄罗斯方块 dfs

首先将7种方块拆成19种方块，然后在进行dfs组合，当然如果给定的N*M不是4的倍数的时候记得直接输出0。

代码如下：

#include <cstdlib>

#include <cstring>

#include <cstdio>

using namespace std;



int N, M, ans, END, G[35][35], many[20];



int mp[20] = {1, 1, 2, 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 6, 6, 7, 7};



char ss[50][6][6] = {

    { "####" }, {"#", "#", "#", "#"}, 

    { "##", "##" }, { "#", "###" }, 

    { "##", "#", "#"}, { "###", "  #" }, 

    { " #", " #", "##" },

    { "  #", "###" }, { "#", "#", "##" },

    { "###", "#" }, { "##", " #", " #" },

    { " #", "###" }, 

    { "#", "##", "#" }, {"###", " #"}, 

    { " #", "##", " #" }, { "##", " ##" },

    { " #", "##", "#" }, { " ##", "##" },

    { "#", "##", " #" }, {"#", "##", " #"}

};



struct Node

{

    char s[6][6];

}e[50];



void pre()

{

    for (int i = 0; i < 19; ++i) {

        memcpy(e[i].s, ss[i], sizeof (ss[0]));

    }

}



bool judge(int x, int y)

{

    if (x < 1 || x > N || y < 1 || y > M) {

        return false;

    }

    else {

        return true;

    }

}



bool Getpos(int &x, int &y)

{

    for (int i = 1; i <= N; ++i) {

        for (int j = 1; j <= M; ++j) {

            if (!G[i][j]) {

                x = i, y = j;

                return true;

            }

        }

    }

    return false;

}



bool OK(int x, int y, int No)

{

    int rec[6][2], cnt = 0, first = 1;

    int _x, _y;

    for (int i = 0; i < 6; ++i) {

        for (int j = 0; j < 6; ++j) {

            if (e[No].s[i][j] == '#') {

                if (first) {

                    _x = i, _y = j;

                    first = 0;

                }

                if (!judge(x+(i-_x), y+(j-_y)) || G[x+(i-_x)][y+(j-_y)] != 0) {

                    return false;

                }

                else {

                    ++cnt;

                    rec[cnt][0] = x + i - _x;

                    rec[cnt][1] = y + j - _y;

                }

            }

        }

    }

    for (int i = 1; i <= cnt; ++i) {

        G[rec[i][0]][rec[i][1]] = 1;

    }

    return true;

}



void erase(int x, int y, int No)

{

    int _x, _y, first = 1;

    for (int i = 0; i < 6; ++i) {

        for (int j = 0; j < 6; ++j) {

            if (e[No].s[i][j] == '#') {

                if (first) {

                    _x = i, _y = j;

                    first = 0;         

                }

                G[x+i-_x][y+j-_y] = 0;

            }

        }

    }

}



void print(int No)

{

    for (int i = 0; i < 6; ++i) {

        for (int j = 0; j < 6; ++j) {

            printf("%c", e[No].s[i][j]);

        }

        puts("");

    }

}



void dfs(int x, int y, int No, int num)

{

    int xx, yy;

    if (!OK(x, y, No)) {

        return;

    }

    --many[mp[No]];

    if (num == END) {

        ++ans;

    }

    else {

        Getpos(xx, yy);

        for (int i = 0; i < 19; ++i) {

            if (many[mp[i]]) {

                dfs(xx, yy, i, num+1);

            }

        }

    }

    erase(x, y, No);

    ++many[mp[No]];

}



int main()

{

    pre();

    int T;

    scanf("%d", &T);

    while (T--) {

        memset(many, 0, sizeof (many));

        ans = 0;

        scanf("%d %d", &N, &M);

        for (int i = 1; i <= 7; ++i) {

            scanf("%d", &many[i]);

        }

        if ((N*M)%4!=0) {

            puts("0");

            continue;

        }

        END = (N*M)/4; 

        for (int k = 0; k < 19; ++k) {

            if (many[mp[k]]) {

                dfs(1, 1, k, 1);

            }

        }

        printf("%d\n", ans);

    }

    return 0;

}