HDU 5749 Colmerauer（单调栈）

Description
Peter有一个n×m的矩阵M. 定义S(a,b)为M的所有大小为a×b的子矩阵的权值和. 一个矩阵的权值是这个矩阵所有鞍点的值的和. 在矩阵中, 一个数在所在行中是唯一的最小值, 在所在列中是唯一的最大值, 则被称为鞍点. 帮助Peter找出所有S(a,b)的值
Input
输入包含多组数据, 第一行包含一个整数T表示测试数据组数. 对于每组数据:
第一行包含两个整数n和m(1≤n,m≤1000)表示矩阵的大小.
接下来n行每行包含m个非负整数, 表示矩阵MM的每一行 (M中每个元素都不超过10^​6).
Output
对于每组数据, 输出一个整数W=
Sample Input
3
2 2
1 1
1 1
3 3
1 2 3
4 5 6
7 8 9
3 3
1 2 1
2 3 1
4 5 2
Sample Output
4
600
215
Solution
考虑每个点的作为鞍点对答案的贡献，首先对每行每列做单调栈，对于每个点s[i][j]就可以得到l,r,u,d四个值表示该点在第j-l列到第j+r列是第i行唯一最小值，在第i-u行到第i+d行是第j列唯一最大值，那么其对答案的贡献就是

对每个点累加上述贡献即可，时间复杂度O(nm)
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
#define mod (1ll<<32)
#define maxn 1111
int T,n,m,s[maxn][maxn],a[maxn],l[maxn],r[maxn];
struct node
{
    int l,r,u,d;
}b[maxn][maxn];
void deal(int *a,int n)
{
    int p,sta[maxn];
    p=0;
    for(int i=1;i<=n;i++)
    {
        if(!p||a[i]>a[sta[p]])sta[++p]=i;
        else 
        {
            while(p&&a[i]<=a[sta[p]])
                r[sta[p]]=i-1,p--;
            sta[++p]=i;
        }
    }
    while(p)r[sta[p]]=n,p--;
    for(int i=n;i>=1;i--)
    {
        if(!p||a[i]>a[sta[p]])sta[++p]=i;
        else 
        {
            while(p&&a[i]<=a[sta[p]])
                l[sta[p]]=i+1,p--;
            sta[++p]=i;
        }
    }
    while(p)l[sta[p]]=1,p--;
}
ll solve(int x,int l,int r,int u,int d)
{
    ll ans1=1ll*(u+1)*(d+1)*(u+d+2)/2;
    ll ans2=1ll*(l+1)*(r+1)*(l+r+2)/2;
    ll ans=ans1*ans2%mod*x%mod;
    return ans;
}
int main()
{
    solve(1,2,3,2,3);
    scanf("%d",&T);
    while(T--)
    {
        ll ans=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                scanf("%d",&s[i][j]);
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)a[j]=s[i][j];
            deal(a,m);
            for(int j=1;j<=m;j++)
                b[i][j].l=l[j],b[i][j].r=r[j];
        }
        for(int j=1;j<=m;j++)
        {
            for(int i=1;i<=n;i++)a[i]=-s[i][j];
            deal(a,n);
            for(int i=1;i<=n;i++)
                b[i][j].u=l[i],b[i][j].d=r[i];
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
            {
                ll temp=solve(s[i][j],j-b[i][j].l,b[i][j].r-j,i-b[i][j].u,b[i][j].d-i);
                ans=(ans+temp)%mod;
            }
        printf("%I64d\n",ans);
    }
    return 0;
}