其他分享
首页 > 其他分享> > P1950 长方形

P1950 长方形

作者:互联网

传送门

以行为单位,统计以第k行为底边的矩形数量,而对于每一行,以每一格为基准计算左方案 li,右方案 ri,高方案 hi。同时为了防止重复计算,左方案 li 为这一列左边满足h值小于等于 hi 的列号(没有的话为0),右方案 ri 为这一列右边第一个满足h值小于hi的列号(如果没有的话就为col+1)。而以这样为基准的每一个小方块的矩阵的数量为(i-li)*(ri-i)*hi。(这样的方法在类似的求矩形数量的题目中都是一种高效的算法)在计算左右方案时为降低复杂度,需要用到单调栈。

#include<iostream>
#define MAXN 1010
using namespace std;
int h[MAXN], r[MAXN], l[MAXN], st[MAXN];
long long ans; char G[MAXN][MAXN];
int main(void)
{
    int n = 0, m = 0, top = 0;
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++)
            cin >> G[i][j];
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
            h[j] = G[i][j] == '*' ? 0 : h[j] + 1;
        for (int j = 1; j <= m; j++)
        {
            while (top > 0 && h[j] < h[st[top]]) {
                r[st[top]] = j;
                top--;
            }
            top++;
            st[top] = j;
        }
        while (top > 0) {
            r[st[top]] = m + 1;
            top--;
        }
        for (int j = m; j >= 1; j--)
        {
            while (top > 0 && h[j] <= h[st[top]]) {
                l[st[top]] = j;
                top--;
            }
            top++;
            st[top] = j;
        }
        while (top > 0) {
            l[st[top]] = 0;
            top--;
        }
        for (int j = 1; j <= m; j++)
            ans += (j - l[j]) * (r[j] - j) * h[j];
    }
    cout << ans;
    return 0;
}

 

标签:P1950,int,top,长方形,st,--,hi,MAXN
来源: https://www.cnblogs.com/xqk0225/p/16079332.html