单调栈之WYT的刷子

好久没更题解了(改题困难的我)

题目描述

WYT有一把巨大的刷子,刷子的宽度为M米,现在WYT要使用这把大刷子去粉刷有N列的栅栏(每列宽度都为1米;每列的高度单位也为米,由输入数据给出)。

使用刷子的规则是:

  • 与地面垂直,从栅栏的底部向上刷
  • 每次刷的宽度为M米(当剩余栅栏宽度不够M米的话,刷子也可以使用,具体看样例2)
  • 对于连续的M列栅栏,刷子从底向上,刷到的高度只能到这M列栅栏的最低高度。

WYT请你回答两个问题:

  • 最少有多少个单位面积不能刷到(单位面积为1平米)
  • 在满足第一问的条件下,最少刷几次?

输入格式

共两行:
第一行两个整数N和M。
第二行共N个整数,表示N列栅栏的高度

输出格式

一行,两个整数,分别为最少剩余的单位面积数量和最少刷的次数。

样例

输入

5 3
5 3 4 4 5

输出

3
2

样例对应解释

高度分别为 5 3 4 4 5 如上:
黄色的方块表示共有3个单位面积没刷上
绿色的框和红色的框表示一共刷了两次。

数据范围与提示

30%的数据:N<=10^3
50%的数据:N<=10^5
100%的数据:1<=N<=10^6, 1<=M<=10^6,N>=M, 每列栅栏的高度<=10^6.

思路

  • 这道题当时真的没有想出来,直接暴力模拟拿了点分,这道题的核心在于维护单调栈,用于求解每块木板向右(向左)延伸的长度,拿向右来说,我们维护单调递增的栈,当栈顶元素的长度小于要放入的元素,则正常入栈,top++,如果栈不为空且栈顶元素(假设为sta[top])大于扫描到的元素(假设为i),对栈顶元素执行出栈操作,然后可得sta[top]向右延伸的距离为r[sta[top]]=i-sta[top],r数组维护向右延伸的长度,同理可求得左边延伸长度;
  • 处理完l和r数组后,对于木板i,其延伸距离为l[i]+r[i]-1,判断其与刷子宽度的大小,大于则可以刷到,用flag标记,小于则不能。然后枚举每一块木板,维护mh数组为第i块木板能被刷到的长度,如果被标记过,则mh[i]=h[i],如果不能,则mh[i]=mh[i-1],用长度和减去可以被刷到的部分即为ans1;
  • 对于每一部分(能被刷到的的长度相等),贪心处理,用其宽度除以刷子宽度,累加得ans2;

附上代码一份

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
long long v[maxn],mh[maxn],sum,r[maxn],l[maxn];
long long n,m;
bool flag[maxn];
inline int read(){
	int s=0;
	char ch=getchar();
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch>='0'&& ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s;
}
long long sta[maxn],top,h[maxn];
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++){
		h[i]=read();
		sum+=h[i];
	}
	for(int i=1;i<=n+1;i++){
		while(top && h[i]<h[sta[top]]){
			r[sta[top]]=i-sta[top];
			top--;
		}
		sta[++top]=i;
	}
	for(int i=n;i>=0;i--){
		while(top && h[i]<h[sta[top]]){
			l[sta[top]]=sta[top]-i;
			top--;
		}
		sta[++top]=i;
	}
	for(int i=1;i<=n;i++){
		if(l[i]+r[i]-1>=m)flag[i]=true;
	}
	for(int i=1;i<=n;i++){
		if(flag[i])mh[i]=h[i];
		else{
			mh[i]=mh[i-1];
		}
	}
	for(int i=n;i>0;i--){
		if(!flag[i])mh[i]=max(mh[i],mh[i+1]);
	sum-=mh[i];
	}
	cout<<sum<<endl;
	long long k=2,ans=0;
	while(k<=n+1){
		int cnt=1;
		while(k<=n+1 && mh[k]==mh[k-1]){
			++cnt;
			++k;
		}
		ans+=cnt/m;
		if(cnt%m)++ans;
		++k;
	}
	cout<<ans<<endl;
}	

原文地址:https://www.cnblogs.com/soda-ma/p/13226798.html