hdu 3531 暴力搜索(细节)

/*
题意:给出一个n*n的和一个m*m的0 1矩阵,求m是否为n的子矩阵

题解:暴力搜索剪枝
看到这道题,没什么思路,能想到的只能有暴力剪枝,然后就是如何剪枝优化速度:
首先是看到矩阵就会容易联想到通过求和来剪枝,因为求和可以在输入的过程中同时
进行,因此不会浪费时间,然后对求出的和进行比较,在n维矩阵中找出所有和等于
m维矩阵的组合,然后再比较对角线上的和是否相等,最后再比较所有的点。

注意:有何多比较的细节问题需要一一去调试
*/
#include <cstdio>
#include <cstring>

#define MAXNUM 1005

int num1[MAXNUM][MAXNUM]; // 记录n维矩阵
int num2[MAXNUM][MAXNUM]; // 记录m维矩阵
int sum1[MAXNUM][MAXNUM]; // 记录n维矩阵的和
int sum2[MAXNUM][MAXNUM]; // 记录m维矩阵的和

int main(void)
{
    int n,m;
    memset(sum1,0,sizeof(sum1));
    memset(sum2,0,sizeof(sum2));

    while (~scanf("%d%d",&n,&m))
    {
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
            {
                scanf("%d",&num1[i][j]);
                sum1[i][j] = num1[i][j] + sum1[i][j-1] + sum1[i-1][j] - sum1[i-1][j-1]; // 求和
            }
        for(int i=1; i<=m; i++)
            for(int j=1; j<=m; j++)
            {
                scanf("%d",&num2[i][j]);
                sum2[i][j] = num2[i][j] + sum2[i][j-1] + sum2[i-1][j] - sum2[i-1][j-1]; // 求和
            }

        bool flag = false;

        for(int i=m; i<=n; i++)
        {
            for(int j=m; j<=n; j++)
            {
                // 判断是否m*m的矩阵的所有点之和是否相等
                if (sum2[m][m] == sum1[i][j] - sum1[i][j-m] - sum1[i-m][j] + sum1[i-m][j-m])
                {
                    int k;
                    // 比较对角线上的sum[i][j]
                    for(k=1; k<=m; k++)
                        if (sum2[k][k] != (sum1[i-m+k][j-m+k] - sum1[i-m][j-m+k] - sum1[i-m+k][j-m] + sum1[i-m][j-m]))
                        {
                            break;
                        }
                    if (k > m)
                    {
                        // 比较所有的点
                        for(k=1; k<=m; k++)
                        {
                            int l;
                            for(l=1; l<=m; l++)
                            {
                                if (num2[k][l] != num1[i-m+k][j-m+l])
                                {
                                    break;
                                }
                            }
                            if (l <= m)
                                break;
                        }
                        if (k > m)
                            flag = true;
                    }
                }
                if (flag)
                    break;
            }
        }

        if (flag)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}

 

posted @ 2014-03-20 23:33  辛力啤  阅读(402)  评论(0编辑  收藏  举报