本周主要学习的是线性数据结构的应用
- 单调栈
- 单调队列
- 尺取法
- 差分和前缀和
并每个点选取了一道题目进行练习。
A单调栈——最大矩形
题目描述:
给一个直方图,求直方图中的最大矩形的面积,直方图宽度均为1。
思路:
要寻找对每个单个的矩阵高度取其向左向右能扩展的宽度,也就是向左向右一直到比其小的数值。如果是暴力求解的话复杂度为O(n^2),很明显会超时,所以我们选用同样能够得到第一个比其小/大的数据结构单调栈。单调栈顾名思义就是一个单调的栈,维护了一个栈自栈顶到栈底的单调性。其时间复杂度为O(n)。
代码解释:
right_min()函数就是从左到右遍历找到右侧第一个比当前i矩形小的值。采用了数组模拟栈的方式,数组中存放的是元素序号i。如果当前栈顶元素即a[sta[r]]比当前元素大,则将之弹出,赋给R[sta[r]],直到小为止,这样R数组内存储的就是第一个小的值。反之则压入。
left_min()函数同理。
坑点
题目中每个h的数据范围是1e9,个数最大范围是1e5,用int会爆!!!所以采用longlong。
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <string.h>
using namespace std;
long long n;
long long ans=0;
long long a[100005];
long long sta[100005];
long long R[100005];
long long L[100005];
void right_min()
{
int l=1,r=0;
for(long long i=0;i<n;i++)
{
while(l<=r&&a[sta[r]]>a[i])
{
R[sta[r]]=i;
r--;
}
sta[++r]=i;
}
}
//7 2 1 4 5 1 3 3
void left_min()
{
int l=0,r=1;
for(long long i=n-1;i>=0;i--)
{
while(l>=r&&a[sta[l]]>a[i])
{
L[sta[l]]=i;
l--;
}
sta[++l]=i;
}
}
int main()
{
while(scanf("%lld",&n)&&n!=0)
{
ans=0;
memset(L,-1,sizeof(L));
memset(a,0,sizeof(a));
memset(sta,0,sizeof(sta));
for(int i=0;i<n;i++)
R[i]=n;
for(int i=0;i<n;i++)
{
scanf("%lld",&a[i]);//storage the number
}
right_min();
memset(sta,0,sizeof(sta));
left_min();
for(int i=0;i<n;i++)
{
ans=max(ans,a[i]*(R[i]-L[i]-1));
}
printf("%lld\n",ans);
}
return 0;
}
/*
7 2 1 4 5 1 3 3
4 1000 1000 1000 1000
0
*/
B TT’s Magic Cat ——差分
题目描述:
给定一个数组,每次操作将[L,R]区间中的每个数均加上c,求q次操作结束后数组中各个元素的值。
思路
如果是暴力求解直接q次循环的话时间复杂度为O(qn),会超时。所以选择将原数组转变为差分数组,将区间修改转变为单点修改。
A[L,R]+=c=>B[L]+=c,B[R+1]-=c
如果r为右端时,只将B[L]+=c即可
B数组的前缀和即为A数组的最终数值。
时间复杂度为O(q+n)
#include <stdio.h>
using namespace std;
int n,q,l,r,c;
long long a[200005];
long long b[200006];
long long sum;
void solve(int l,int r,int c)
{
b[l]+=c;
if(r!=n) b[r+1]-=c;
}
int main()
{
scanf("%d %d",&n,&q);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
a[0]=0;
for(int i=1;i<=n;i++)
{
b[i]=a[i]-a[i-1];
}
for(int i=0;i<q;i++)
{
scanf("%d %d %d",&l,&r,&c);
solve(l,r,c);
}
for(int i=1;i<=n;i++)
{
sum+=b[i];
if(i==n)
printf("%lld",sum);
else printf("%lld ",sum);
}
return 0;
}
C 平衡字符串——尺取法
题目描述
替换字符串s的最小子串使之成为 ‘Q’, ‘W’, ‘E’, ‘R’ 个数相等的平衡队列。求最小子串的长度。
思路
首先分析题意我们需要得到的结果是一个连续的区间,如果暴力求解的话复杂度为O(n^2),既然连续我们可以选择用尺取法。题目也满足[L,R] 满足要求,则L++,不满足要求,则R++。
判断该区间是否满足要求:
- 用sum1, sum2, sum3, sum4分别记录不包含区间 [L, R] 这一段时,字符 ‘Q’, ‘W’, ‘E’, ‘R’ 的个数
- 先通过替换使 4 类字符数量一致,再判断剩余空闲位置是否 为 4 的倍数
- MAX = max(sum1, sum2, sum3, sum4) • TOTAL = R – L + 1
- FREE = TOTAL [(MAX-sum1)+(MAX-sum2)+(MAX-sum3)+(MAX-sum4)]
- 若 FREE ≥0 且为 4 的倍数,则满足要求;否则不满足
在得到区间长度的时候利用前缀和进行表示,可以很轻易地判断出去除当前区间后的每个字符的个数。
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
char c[100005];
int Q[100005],W[100005],E[100005],R[100005];
int ans=1000000;
int l,r;
void add(int Q[100005],int W[100005],int E[100005],int R[100005],int i)
{//计算前缀和
Q[i]=Q[i-1]+1;
W[i]=W[i-1];
E[i]=E[i-1];
R[i]=R[i-1];
}
int main()
{
cin>>c;
if(c[0]=='Q') Q[0]=1;
if(c[0]=='W') W[0]=1;
if(c[0]=='E') E[0]=1;
if(c[0]=='R') R[0]=1;
for(int i=1;i<strlen(c);i++)
{//利用前缀和,便于后期判断区间内字母个数
switch(c[i])
{
case 'Q':
add(Q,W,E,R,i);
break;
case 'W':
add(W,Q,E,R,i);
break;
case 'E':
add(E,Q,W,R,i);
break;
case 'R':
add(R,Q,W,E,i);
break;
default: break;
}
}
int l=0,r=0;
int n=strlen(c);
while(l<n&&r<n)
{
int sum1,sum2,sum3,sum4=0;
if(l==0)
{
sum1=Q[n-1]-Q[r];
sum2=W[n-1]-W[r];
sum3=E[n-1]-E[r];
sum4=R[n-1]-R[r];
}
else
{
sum1=Q[n-1]-Q[r]+Q[l-1];
sum2=W[n-1]-W[r]+W[l-1];
sum3=E[n-1]-E[r]+E[l-1];
sum4=R[n-1]-R[r]+R[l-1];
}
int maxx=max(max(sum1,sum2),max(sum3,sum4));
int len=r-l+1;
int total=len-(maxx-sum1)-(maxx-sum2)-(maxx-sum3)-(maxx-sum4);
if(total>=0&&total%4==0)
{
ans=min(ans,len);
l++;
}
else r++;
}
printf("%d",ans);
return 0;
}
D 滑动窗口——单调队列
题目描述:
给定一个数组和窗口,求在窗口从左往右滑的时候,每次窗口内数的最大值和最小值分别是多少
思路:
查找窗口内的最小值是一个局部的概念,可以维护一个单调队列,求最小值时队首到队尾递增,最大值则反之。队列中的元素均属于当前窗口,很容易即可得到窗口内的最值。当元素不属于当前窗口时,将队首元素弹出即可
#include <stdio.h>
#include <iostream>
using namespace std;
int n=0,k=0;
int a[1000005];
int q[1000005];//queue
int minnum[1000005];
int maxnum[1000005];
void min_num()//一个头最小的队列
{
int l=1,r=0;
for(int i=0;i<k-1;i++)
{
while(r>=l&&a[q[r]]>a[i])
{
r--;//弹出
}
q[++r]=i;//i压入
}
for(int i=k-1;i<n;i++)
{
while(r>=l&&a[q[r]]>a[i])
{
r--;//弹出
}
q[++r]=i;//i压入
if(q[r]-q[l]+1>k) l++;//保证窗口长度
minnum[i]=q[l];
if(i==n-1) printf("%d\n",a[minnum[i]]);
else printf("%d ",a[minnum[i]]);
}
}
void max_num()
{
int l=1,r=0;
for(int i=0;i<k-1;i++)
{
while(r>=l&&a[q[r]]<a[i])
{
r--;//弹出
}
q[++r]=i;//i压入
}
for(int i=k-1;i<n;i++)
{
while(r>=l&&a[q[r]]<a[i])
{
r--;//弹出
}
q[++r]=i;//i压入
if(q[r]-q[l]+1>k) l++;//保证窗口长度
maxnum[i]=q[l];
if(i==n-1) printf("%d",a[maxnum[i]]);
else printf("%d ",a[maxnum[i]]);
}
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
min_num();
for(int i=0;i<n;i++)
q[i]=0;
max_num();
return 0;
}