答案:
#include <bits/stdc++.h>
#include <string.h>
/* '/'的用法:
\是行连续的意思,有这个符号,
就是下一行是接着上一行的意思,
就是下一行和上一行从语法上来说是一行,
只是分成了多行来写。
如果写在一行,可能太长了,
不美观,所以用\分割一下子 */
#define FAST_IO \
ios::sync_with_stdio(false); \
cin.tie(0); \
cout.tie(0);
using namespace std;
int main()
{
//cin cout的加速版
FAST_IO;
int num, cap, n;
cin >> num >> cap >> n;
/* n--的理由:
在这个while循环的条件判断里面
必须写n--,不准写--n,
倘若写了--n则会少循环一次 */
while (n--)
{
stack<int> st;
int flag = 1;
int sign = 1;
for (int i = 0; i < num; i++)
{
int x;
cin >> x;
if (x == sign)
{
sign++;
while (!st.empty() && st.top() == sign)
{
st.pop();
sign++;
}
}
else
{
st.push(x);
if (st.size() > cap)
flag = 0;
}
}
if (flag && sign == num + 1)
cout << "YES";
else
cout << "NO";
if (n)
cout << endl;
}
return 0;
}
回顾:
算法:一个一个去读取所输入的发货顺序,设置一个变量用来表示当前所需要的颜料序号(用sign表示),当所读取到的颜料序号与sign一致时,sign++,之后先不要急着去读取下一个颜料序号,题目中也讲了“当一种颜色装填完以后,先看看货架顶端的一箱是不是下一个要装填的颜色,如果是就取下来装填,否则去工厂里再搬一箱过来。”,因而sign++之后应该接着判断栈顶元素的序号是否与sign一致,若一致则继续sign++然后重复上述步骤,直到栈空或栈顶元素与sign不一致为止,此时再继续读取下一个颜料序号;当所读取到的颜料序号与sign不一致时,要将该颜料序号入栈,入栈之后再判断栈长度是否超过了规定长度,若超过了,则判定为失败。
细节:读取很重要(在处理一组发货顺序时,一定要恰好读取num个数,不要多读,也不要少读,否则都会影响结果的正确性)因而最好一个for循环,循环num次,且每次都只cin一次,这样解决了读取数量这个问题。
判断:首先在读取过程中,如果存在栈超额的情况,那么一定是“NO”的。但这并不是唯一的情况,在一开始写代码的时候,我用一个数组来表示某种序号的颜料是否已经被读取,以此来处理栈顶与sign不匹配,且sign所指颜料已经在栈中的情况。相比于一开始的想法,用一个sign来表示当前所需的颜料序号这一方法更节省空间。而上述所说明的问题就会导致这一组序号都输入完之后,sign所指颜料并非num+1,即并没有完全上好色,所以这种情况也要考虑进去。