【bzoj2003】[Hnoi2010]Matrix 矩阵

题目链接

Description

Input

第一行包含三个正整数N M P表示矩阵的行数列数以及每个数的范围，接下来N行每行包含M个非负整数，其中第i行第j个数表示以格子(i,j)为右下角的2*2子矩阵中的数的和。保证第一行与第一列的数均为0，且每个和都不超过4(P-1)。

Output

包含N行，每行M个整数，描述你求出的矩阵，相邻的整数用空格分开。（行末不要有多余空格）

Sample Input

3 3 3

0 0 0

0 4 5

0 5 3

Sample Output

0 0 2

2 2 1

1 0 0

HINT

$1\leq N,M\leq200$
$1<P\le 10$

题解

意识流。。
首先构造一个矩阵$C[M][N]$:

$$C[i][j] =S[i][j] - C[i-1][j]-C[i][j-1]-C[i-1][j-1]$$
$$\text{特殊的}：C[i][0] = C[0][j] = 0$$
然后考虑第一行和第一列的数如果确定了就可以确定整个矩阵。
设最终的矩阵为 $A[M][N]$其中第一行和第一列已经确定。
那么：
$$A[i][j] = C[i][j]+A[0][0]\times f(i+j+1)+A[i][0]\times f(j)+A[0][j]\times f(i)$$
$$\text{其中}f(x) = \begin{cases} -1 & x\%2 = 1\\ 1 & x\%2 = 0 \end{cases}$$
然后我们要做的就是从小到大枚举第一行的值，然后看第一列有没有可行解，若有就可以退出搜索了，因为此时一定是字典序最小的解。
具体实现是通过计算 $A[i][0]$可以填写的数（使第i行所有数都在范围内）的最大值和最小值。
考虑已经填写好 $A[0][0]\dots A[0][j]$，使 $A[i][[j]$在范围内的 $A[i][0]$的极值：
$C[i][j]+A[0][0]\times f(i+j+1)+A[0][j]\times f(i)+x\times f(j) = 0$
$C[i][j]+A[0][0]\times f(i+j+1)+A[0][j]\times f(i)+y\times f(j) = p-1$
那么对于 $A[i][j]$来说 $A[i][0]$的取值是在 $[min(x,y),max(x,y)]$间的，将这些集合取交就是 $A[i][j]$的真实范围了。若在过程中集合变为空集，就可以剪枝。

#include<bits/stdc++.h>
using namespace std;

inline int read(){
    int x = 0, f = 1; char c = getchar();
    while(!isdigit(c)) { if(c == '-') f = -1; c = getchar(); }
    while(isdigit(c)) { x = x * 10 + c - '0'; c = getchar(); }
    return x * f;
}

const int N = 200 + 10;

int a[N][N], c[N][N], n, m, p;
int up[N][N], dn[N][N];


void init(){
    n = read(); m = read(); p = read();
    for(int i = 0; i < n; i++)
    for(int j = 0; j < m; j++){
        int x = read();
        if(i && j) c[i][j] = x - c[i-1][j] - c[i][j-1] - c[i-1][j-1];
        up[i][j] = p - 1;
    }
}

inline int f(int x){return x & 1 ? -1 : 1;}

bool dfs(int j){
    if(j == m) return true;
    for(int k = 0; k < p; k++){
        a[0][j] = k;
        bool ok = true;
        for(int i = 1; i < n; i++){
            int x = (c[i][j] + f(i+j+1) * a[0][0] + f(i)*a[0][j]) * (-f(j));
            int y = (c[i][j] + f(i+j+1) * a[0][0] + f(i)*a[0][j] - (p-1)) * (-f(j));
            if(x > y) swap(x, y);
            dn[i][j] = max(dn[i][j-1], x);
            up[i][j] = min(up[i][j-1], y);
            if(dn[i][j] > up[i][j]){
                ok = false;
                break;
            }
        }
        if(ok)
            if(dfs(j+1)) return true;
    }
    return false;
}

void work(){
    for(a[0][0] = 0; a[0][0] < p && !dfs(1); a[0][0]++);
    for(int i = 1; i < n; i++) a[i][0] = dn[i][m-1];
    for(int i = 0; i < n; i++)
    for(int j = 0; j < m; j++)
        printf("%d%s", f(j)*a[i][0]+f(i)*a[0][j]+f(i+j+1)*a[0][0]+c[i][j], j+1 == m ? "\n" : " ");
}

int main(){
    init();
    work();
    return 0;
}