目录
题目
Description
emmm,还是北湖深坑,不用惊喜,不用意外
我们继续用石头填!
北湖的地面依旧是一维的,每一块宽度都为1,高度是非负整数,用一个数组来表示。
还是提供不限量的 规格的石头。
但是这一次是 来填坑,他有很强烈的强迫症,所有的石头只能水平摆放(宽为2,高为1)。
问这样是否可以将北湖填平。(所有地面到达同一高度即为填平)
Input
样例有多组输入至文件末尾;
每组用例占两行;
第一行输入1个整数 表示北湖地面总宽度;
第二行输入 个整数 ,用空格间隔,表示地面高度。
Output
若能填平则输出“YES”,否则输出“NO”。
思路
- 使用一个栈来模拟填坑的过程,栈中存储当前填坑的高度。
- 遍历地面高度数组,如果当前高度小于栈顶高度,则将当前高度入栈。
- 如果当前高度等于栈顶高度,则将栈顶高度出栈。
- 如果当前高度大于栈顶高度,则无法填平北湖,将flag标记为0。
- 遍历结束后,如果flag为0,则输出"NO";否则,判断栈是否为空或栈中只有一个元素且该元素为最大高度,如果是,则输出"YES",否则输出"NO"。
注意事项
-
输入数据的范围:在定义地面高度数组时,最好使用 long long 类型。
-
栈的使用:题目中需要使用栈来模拟填坑的过程。在入栈和出栈时,需要注意栈是否为空,以及栈顶元素的判断。
-
地面高度的更新:在遍历地面高度数组时,需要及时更新地面的最大高度。可以使用一个变量 maxh 来记录当前的最大高度,每次读取新的地面高度时,将其与 maxh 进行比较,并更新 maxh 的值。
-
输出结果的判断:根据题目要求,需要判断是否能填平北湖。如果不能填平,则输出 "NO";如果能填平,则需要判断栈是否为空或栈中只有一个元素且该元素为最大高度,如果是,则输出 "YES",否则输出 "NO"。
-
注意处理多组输入:题目要求多组输入,所以需要使用一个循环来读取输入,直到文件末尾。(scanf("%lld", &n) != EOF)
-
注意清空栈和数组:每次读取新的输入前,需要清空栈和数组,以便处理新的输入。
C++完整代码(含详细注释)
#include <iostream>
#include <vector>
#include <stack>
#include <algorithm>
using namespace std;
int main(){
long long n; // 输入的地面高度个数
while(scanf("%lld", &n) != EOF){ // 循环读取输入,直到文件末尾
stack<long long> st; // 定义一个栈来模拟填坑的过程
vector<long long> h(n, 0); // 定义一个大小为n的数组h来存储地面高度
long long maxh = 0; // 地面的最大高度
int flag = 1; // 是否能填平北湖的标记,初始值为1
for (long long i = 0; i < n; i++) { // 遍历地面高度数组
scanf("%lld", &h[i]); // 读取一个地面高度
maxh = max(maxh, h[i]); // 更新地面的最大高度
if (st.empty() || h[i] < st.top()) { // 如果当前高度小于栈顶高度,则将当前高度入栈
st.push(h[i]);
}
else if (h[i] == st.top()) { // 如果当前高度等于栈顶高度,则将栈顶高度出栈
st.pop();
}
else if(h[i]>st.top()){ // 如果当前高度大于栈顶高度,则无法填平北湖,将flag标记为0
flag = 0;
}
}
if(flag==0){ // 如果flag为0,则输出"NO"
cout << "NO" << endl;
}
else if(st.empty() || (st.size() == 1 && st.top() == maxh)) { // 如果栈为空或栈中只有一个元素且该元素为最大高度,则输出"YES"
cout << "YES" << endl;
}
else { // 否则输出"NO"
cout << "NO" << endl;
}
}
return 0;
}