[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<=N,M<=200

1<P<=10

首先可以推出,一旦A的第一行与第一列已经确定,那么我们就可以按部就班地
算出A的所有元素的值了。这样一来需要决策的元素个数就从$200^{2}$变成了399
首先不看元素在0到P-1范围内的限制,设A(1,k)和A(k,1)都为0,并计算将A补全。
(此时A中可能有负数或者大于等于P的整数)
这样A中有些值不符合条件，调整A矩阵
当我们改变A(1,1)+=k时,
要把A调整成合法矩阵,事实上只需把i+j为偶数的A(i,j)
全部增加 k,i+j为奇数的 A(i,j)全部减去 k 即可
当我们把A(1,i)增加k时,要把A调整成合法,只需要把第i列奇数位加k,偶数位减k即可。
我们只需要搜索A的第一行,每当确定一个元素,就可以更新A(j,1)的取值范围。
根据公式,除了A(1,1)以外,A(i,1)的取值是互不影响的。
不断更新第一列每个元素的可以取值的范围,一旦出现有某个元素下界大于上界的情况就剪枝
显然这样搜索的字典序是最优的

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 int l[][],r[][],a[][],s[][],c[][],n,m,p;
 int pd(int x)
 {
   if (x&) return -;
   return ;
 }
 int get_num(int x,int y)
 {
   return c[x][y]-pd(x+y)*a[][]-pd(x)*a[][y]-pd(y)*a[x][];
 }
 bool dfs(int x)
 {int k,i,j,ok;
   if (x>m) return ;
   for (k=;k<p;k++)
     {
       ok=;
       a[][x]=k;
       for (i=;i<=n;i++)
     {
       int rl=(c[i][x]-pd(x+i)*a[][]-pd(i)*a[][x])*(pd(x));
       int rr=(c[i][x]-pd(x+i)*a[][]-pd(i)*a[][x]-(p-))*(pd(x));
       if (rl>rr) swap(rl,rr);
       l[i][x]=max(l[i][x-],rl);
       r[i][x]=min(r[i][x-],rr);
       if (l[i][x]>r[i][x])
         {
           ok=;
           break;
         }
     }
     if (ok&&dfs(x+)) return ;
     }
   return ;
 }
 int main()
 {int i,j,k;
   cin>>n>>m>>p;
   for (i=;i<=n;i++)
     {
       for (j=;j<=m;j++)
     {
       scanf("%d",&s[i][j]);
       l[i][j]=;r[i][j]=p-;
     }
     }
   for (i=;i<=n;i++)
     {
       for (j=;j<=m;j++)
     {
       c[i][j]=s[i][j]-c[i-][j]-c[i][j-]-c[i-][j-];
     }
     }
   for (i=;i<p;i++)
     {
       a[][]=i;
       if (dfs())
     {
       for (j=;j<=n;j++)
         {
           a[j][]=l[j][m];
         }
       for (j=;j<=n;j++)
         {
           for (k=;k<=m;k++)
         a[j][k]=get_num(j,k);
         }
       break;
     }
     }
   for (i=;i<=n;i++)
     {
       for (j=;j<m;j++)
     {
       printf("%d ",a[i][j]);
     }
       printf("%d",a[i][m]);
       printf("\n");
     }
 }

首先可以推出,一旦 A 的第一行与第一列已经确定,那么我们就可以按部就班地
算出 A 的所有元素的值了。这样一来需要决策的元素个数就从 200 2 变成了 399。
为了方便计算,我们试图建立 A i,j 与第一行或第一列之间的直接等式关系。
以下是一个较易理解的建立等量关系的方法:
首先不看元素在 0 到 P-1 范围内的限制,设 A 1,k 和 A k,1 都为 0,并计算将 A 补全。
(此时 A 中可能有负数或者大于等于 P 的整数)
S:
0 0 0
0 4 5
0 5 3A:
0 0 0
0 4 1
0 1 -3
当我们改变 A 1,1 +=k 时,
要把 A 调整成合法矩阵,事实上只需把 i+j 为偶数的 A i,j
全部增加 k,i+j 为奇数的 A i,j 全部减去 k 即可。
A’: A 1,1 +=1
1 -1 1
-1 5 0
1 0 -2
当我们把 A 1,i 增加 k 时,要把 A 调整成合法,只需要把第 i 列奇数位加 k,偶数位
减 k 即可。
A’: A 1,2 +=1
0 1 0
0 3 1
0 2 -3
修改 A i,1 类似。
设通过令第一行第一列都为 0 得到的矩阵为 C,
那么有公式:
A i,j = C i,j + (-1) i+j-2 A 1,1 + (-1) i-1 A 1,j + (-1) j-1 A i,1
(i>1,j>1)
接下来只需要考虑对第一行第一列进行决策即可。
前 30%的数据可以搜索解决。
另 30%实际上是一个 2-Sat.问题。参见 SGU 307 Cipher。
对于 100%的数据并没有很好的方法。
研究 30%的搜索方法,如果加入较强的剪枝,一般出解是很快的。
我们只需要搜索 A 的第一行,每当确定一个元素,就可以更新 A j,1 的取值范围。
根据公式,除了 A 1,1 以外,A i,1 的取值是互不影响的。
不断更新第一列每个元素的可以取值的范围,一旦出现有某个元素下界大于上界的
情况就剪枝。
显然这样搜索的字典序是最优的。
事实证明这个剪枝效率非常高。