1067. 最匹配的矩阵

题目描述

给定一个m*n的矩阵A和r*s的矩阵B,其中0 < r < = m,0 < s < = n，A、B所有元素值都是小于100的正整数。求A中一个大小为r*s的子矩阵C，使得B和C的对应元素差值的绝对值之和最小，这时称C为最匹配的矩阵。如果有多个子矩阵同时满足条件，选择子矩阵左上角元素行号小者，行号相同时，选择列号小者。

输入

第一行是m和n，以一个空格分开。
之后m行每行有n个整数，表示A矩阵中的各行，数与数之间以一个空格分开。
第m+2行为r和s，以一个空格分开。
之后r行每行有s个整数，表示B矩阵中的各行，数与数之间以一个空格分开。

输出

输出矩阵C，一共r行，每行s个整数，整数之间以一个空格分开。

样例输入

3 3
3 4 5
5 3 4
8 2 4
2 2
7 3
4 9

样例输出

4 5
3 4

数据范围限制

1< = m < =100,1< = n < = 100

代码如下

#include <bits/stdc++.h>
using namespace std; 
int a[100][100], b[100][100];
int main(){
    int m, n, r, s, i, j;

    scanf("%d%d", &m, &n);
    for(i=0; i<m; i++)
        for(j=0; j<n; j++)
            scanf("%d", &a[i][j]);

    scanf("%d%d", &r, &s);
    for(i=0; i<r; i++)
        for(j=0; j<s; j++)
            scanf("%d", &b[i][j]);

    int drow, dcol, min;
    min = INT_MAX;
    for(i=0; i<m-r+1; i++)
        for(j=0; j<n-s+1; j++) {
            int k, l, sum;
            sum = 0;
            for(k=0; k<r; k++)
                for(l=0; l<s; l++)
                    sum += abs(a[i + k][j +l] - b[k][l]);
            if(sum < min) {
                drow = i, dcol = j;
                min = sum;
            }
        }

    for(i=0; i<r; i++) {
        for(j=0; j<s; j++)
            printf("%d ", a[drow + i][dcol + j]);
        printf("\n");
    }

    return 0;
}