L2-032 彩虹瓶 (25 分)

L2-032 彩虹瓶 (25 分)

题目链接

思路

直接用栈模拟就可以了，没有特别需要注意的，具体见代码。

模拟赛做了一遍，卡在超过m才是判错；现在又做了一遍，还是卡在超过m才是判错。呜呜呜

代码

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f3f3f
#define pb push_back
using namespace std;
typedef pair<int, int> P;
const int N = 1e5 + 10;
typedef long long ll;

struct node
{
    int id;
    int data;
};

stack<int> st;

int main()
{
    int n, m, k;
    cin >> n >> m >> k;
    while(k--)
    {
        while(st.size())
            st.pop();
        bool flag = 0;
        int exp = 1;
        for(int i = 0, tmp; i < n; i++)
        {
            cin >> tmp;
            if(exp == tmp)
            {
                exp++;
                while(st.size() && exp == st.top())
                {
                    st.pop();
                    exp++;
                }
            }
            else
            {
                st.push(tmp);
            }
            if(st.size() > m)
            {
                flag = 1;
            }
        }
        while(st.size() && exp == st.top())
        {
            st.pop();
            exp++;
        }
        if(flag || st.size())
        {
            cout << "NO" << endl;
        }
        else
        {
            cout << "YES" << endl;
        }
    }
    return 0;
}