【单调栈 && 子矩阵】ACM-ICPC 2018 南京赛区网络预赛 B. The writing on the wall

Step1 Problem:

给你 n*m 的矩阵，里面有 k 个黑块，给你 k 个黑块的下标。
输出所有不包含黑块的子矩阵的个数和
数据范围：
1 <= n <= 1e5, 1 <= m <= 100, 0 <= k <= 1e5.

Step2 Ideas:

前置技能：以 (x, y) 为右下角的子矩阵个数 = sum{ f(i) }, i = 1, 2, 3, …, y. 其中 f(i) 代表 x - max{ black(j) }, j = i, i+1, …, y. 其中 black(j) 代表第 j 列最大行下标的黑块.
我们只要 O(1) 的求出以每个下标为右下角的子矩阵个数。
时间复杂度就是 O(n*m).
核心就是我们如何 O(1) 的求出每个下标为右下角的子矩阵个数：
每一行，我们可以维护一个 递增的 单调栈。
每一行，我们维护每个位置的答案。
具体看代码就知道如何 O(1) 求出了。

Step3 Code:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e5+5;
const int M = 105;
struct node
{
    int col, row;
};
bool Map[N][M];
int black[M];
ll sum[M];
int main()
{
    int T, Case = 1, n, m, k, u, v;
    scanf("%d", &T);
    while(T--)
    {
        memset(Map, 0, sizeof(Map));
        scanf("%d %d %d", &n, &m, &k);
        for(int i = 1; i <= k; i++)
        {
            scanf("%d %d", &u, &v);
            Map[u][v] = 1;
        }
        memset(black, 0, sizeof(black));
        ll ans = 0;
        for(int i = 1; i <= n; i++)
        {
            stack<node> q;//维护单调递增的栈，用来记录黑块的位置
            q.push((node){0, n+1});//添加一个边界，好求一点。
            sum[0] = 0;
            for(int j = 1; j <= m; j++)
            {
                if(Map[i][j])
                    black[j] = max(black[j], i);//更新这一列最大行黑块
                while(q.top().row <= black[j]) {//弹出没有效果的黑块
                    q.pop();
                }
                sum[j] = sum[q.top().col] + (i-black[j])*(j - q.top().col);//O(1) 求出当前位置的答案
                q.push((node){j, black[j]});//入栈
                ans += sum[j];//叠加起来
            }
        }
        printf("Case #%d: %lld\n", Case++, ans);
    }
    return 0;
}