补题 Problem E: 积木积水 分治算法

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/hnust_V/article/details/51746361
Problem E: 积木积水
Description
现有一堆边长为1的已经放置好的积木,小明(对的,你没看错,的确是陪伴我们成长的那个小明)想知道当下雨天来时会有多少积水。小明又是如此地喜欢二次元,于是他把这个三维的现实问题简化成二维的问题。设雨量无穷、积木不透水、积木间无缝连接,问在这个二次元的世界里,已放置好的积木会有多少单位的积水量?

这里写图片描述

Input
第一行包含一个整数T(T≤100),表示接下来的测试样例个数。 每个测试样例有两行组成: 第一行包含一个整数N(N≤1e6),表示积木的列数; 第二行包含N个整数Ai(Ai≤1e6),表示第i列积木的个数。

Output
每个样例输出一行,包含一个整数,为题目所求。

Sample Input
1
11
6 2 2 4 2 0 3 4 4 5 1
Sample Output
19
HINT

来源: http://gdutcode.sinaapp.com/problem.php?cid=1039&pid=4
#include <bits/stdc++.h>
using namespace std;
#define MAXN 1000000
int A[MAXN+1];
long long ans;
void CAL(int lef,int rig)  //产生W结构时且W中点大于任一端点则分治,否则直接计算
{
    int L_max=A[lef],R_max=A[rig],MAX= -20,LOC_max=0;
    for(int i=lef+1;i<=rig-1;i++)
        if(MAX<A[i]) {  MAX=A[i];LOC_max=i;   }
    LOC_max = (A[LOC_max]>=L_max||A[LOC_max]>=R_max)?LOC_max:0;
    if(LOC_max) {CAL(lef,LOC_max);CAL(LOC_max,rig);return;}
    else
    {
        int ave=min(A[lef],A[rig]);
        for(int i=lef+1;i<=rig-1;i++) ans+=(ave-A[i]);
    }
    return;
}
int main()
{
    int T;scanf("%d",&T);
    for(int i=1,n,j;i<=T&&scanf("%d",&n);i++)
    {
        int lef,rig;
        for(j=1;j<=n;j++) scanf("%d",&A[j]);
        for(j=1;j<=n&&A[j]<=A[j+1];j++); lef=j;//去掉左端递增项
        for(j=n;j>=1&&A[j]>=A[j-1];j--); rig=j;//去掉右端递减项
        ans=0;CAL(lef,rig);
        printf("%I64d\n",ans);
    }
    return 0;
}
展开阅读全文

没有更多推荐了,返回首页