二维差分约束是一种便于区间更新的存储方式。转载出处
什么是二维差分
我们有一个矩阵,如下图所示。
根据二维前缀和表示的是右上角矩形的和,由于差分只涉及前面相邻的数(由一维可以推出),并且由前面范围的数相加得到这个位置的数。那么类比二维前缀和和一维差分,可以简单推测出二维差分的公式
如何从差分矩阵得到原矩阵呢?可以参考下面公式
但本蒟蒻没看懂, 就写程序跑了一遍上面的公式发现没有得到原矩阵, 在csdn上翻了翻
找到了一个能跑回原矩阵的公式
a[i][j] = a[i - 1][j] + a[i][j - 1] - a[i - 1][j - 1] + b[i][j];
(如果交换一下a[i][j] 与 p[i][j]的话,他喵的就是上面的第一个公式)。
举例
比如,我们有一个矩阵 a,如下所示:
1 2 4 3
5 1 2 4
6 3 5 9
那么对应的二维差分矩阵 p 如下:
1 1 2 -1
4 -5 -1 3
1 1 1 2
应用
如果我们要在左上角是 (x1,y1),右下角是 (x2,y2) 的矩形区间每个值都 +a,如下图所示
在我们要的区间开始位置(x1,y1)处 +c,根据前缀和的性质,那么它影响的就是整个黄色部分,多影响了两个蓝色部分,所以在两个蓝色部分 -c 消除 +c 的影响,而两个蓝色部分重叠的绿色部分多了个 -c 的影响,所以绿色部分 +c 消除影响。所以对应的计算方法如下:
diff[x1][y1] += c;
diff[x1][y2+1] -=c;
diff[x2+1][y1] -=c;
diff[x2+1][y2+1] += c;
2020ICPC·小米 网络选拔赛第一场 J题
题意如下:
给定一个n * m的矩阵A,你每次可以选择A中一个a * b大小子矩阵,让子矩阵中所有元素自减1,经过很多次操作后,问是否能让矩阵A变为全零矩阵(即矩阵中所有元素全为0)如果能输出笑脸(打不出来),否则输出QAQ
输入: 给定T组数据,每组数据输入n, m, a, b,以及一个n * m 的矩阵
输出如题意所示
解题思路
用差分矩阵进行存储与更新,通过判断更新后其差分矩阵元素是否小于零来进行标记,通过标记来进行输出
详细步骤
1设计好差分矩阵的存储和更新函数
1)存储函数
void init(int i, int j)
{
b[i][j] = a[i][j] - a[i - 1][j] - a[i][j - 1] + a[i - 1][j - 1];
}
2)更新函数
void add(int x, int y, int u)
{
b[x][y] += u;
b[x + ai ][y] -= u;
b[x][y + bi ] -= u;
b[x + ai][y + bi] += u;
}
1.输入矩阵值a[i][j]的同时,保存好其差分矩阵调用init(i,j)
2.在遍历差分矩阵的值b[i][j] .( i范围[1, n - ai + 1], j 范围[1, m - bi + 1])
1)如果b[i][j]>0,则调用函数add(i, j, -b[i][j]);
2) 如果b[i][j] <0, flag = 1;
3.如果flag == 1 输出 笑脸, 否则输出 QAQ .
AC代码
#include<bits/stdc++.h>
using namespace std;
int a[2000][2000], b[2000][2000];
int t, n, m, ai, bi;
void init(int i, int j)
{
b[i][j] = a[i][j] - a[i - 1][j] - a[i][j - 1] + a[i - 1][j - 1];
}
void add(int x, int y, int u)
{
b[x][y] += u;
b[x + ai ][y] -= u;
b[x][y + bi ] -= u;
b[x + ai][y + bi] += u;
}
int main()
{
cin >> t;
while (t--)
{
memset(b, 0, sizeof b);
cin >> n >> m >> ai >> bi;
for (int i = 1; i <= n; ++i) {
for(int j = 1; j <= m; ++j) {
cin >> a[i][j];
init(i, j);
}
}
//cout << "sfd\n";
int flag = 0;
for (int i = 1; i <= n - ai + 1; ++i) {
for(int j = 1; j <= m - bi + 1; ++j) {
if (b[i][j] < 0) {
flag = 1;
break;
}
if (b[i][j] > 0) {
add(i, j, -b[i][j]);
}
} if(flag == 1) break;
}
if (flag == 0) {
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= m; ++j) {
if (b[i][j] > 0) {
flag = 1;
break;
}
}
if (flag == 1) break;
}
}
if (flag == 1) cout << "QAQ\n";
else cout << "^_^\n";
}
return 0;
}