原题 扔盘子
有一口井,井的高度为N,每隔1个单位它的宽度有变化。现在从井口往下面扔圆盘,如果圆盘的宽度大于井在某个高度的宽度,则圆盘被卡住(恰好等于的话会下去)。
盘子有几种命运:1、掉到井底。2、被卡住。3、落到别的盘子上方。
盘子的高度也是单位高度。给定井的宽度和每个盘子的宽度,求最终落到井内的盘子数量。
如图井和盘子信息如下:
井:5 6 4 3 6 2 3
盘子:2 3 5 2 4
最终有4个盘子落在井内
Input
第1行:2个数N, M中间用空格分隔,N为井的深度,M为盘子的数量(1 <= N, M <= 50000)。
第2 - N + 1行,每行1个数,对应井的宽度Wi(1 <= Wi <= 10^9)。
第N + 2 - N + M + 1行,每行1个数,对应盘子的宽度Di(1 <= Di <= 10^9)
Output
输出最终落到井内的盘子数量。
解题思路:
使用栈,避免多次循环。
井从上往下,大小应该依次递减,如果上面的井小,无论下面的井有多大,都和上面的一样功能。
#include<cstdio>
#include<algorithm>
#include<stack>
using namespace std;
const int maxn = 50005;
stack<int> sta;
int a[maxn];
int main(){
int n,m,x;
scanf("%d%d",&n,&m);
scanf("%d",&a[0]);
sta.push(a[0]); //将第一个推进栈
for(int i = 1;i < n;i++){
scanf("%d",&a[i]);
if(a[i] >= a[i-1]){ //无论后面的井口多大,都看成上面的大小
a[i] = a[i-1];
}
sta.push(a[i]); // 将所有的井口进栈
}
int sum = 0;
for(int i = 0;i < m;i++){
scanf("%d",&x);
while(!sta.empty()){
if(x <= sta.top()){ //如果盘子可以放进最小的井口
sum++; // 数量加1
sta.pop(); // 将其推出栈
break;
}
sta.pop(); //如果放不下,将所有井口小的推出栈
}
}
printf("%d\n",sum);
return 0;
}