L2-037 包装机

题意

依次按下不同按钮将包装移动，输出取出的包装

思路

• 1.利用queue队列存放轨道的包装，栈stack存放筐内的包装
• 2.当按钮为0时启动机械臂，从筐中抓取，判断筐内是否有包装
• 3.当按钮为轨道时，判断筐内是否还可以放下，将对应的包装放入筐中

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
queue<char>qu[N];//存放传送带上的包装
stack<char>st;//栈先进后出
int main()
{
    int n,m,maxn;
    cin>>n>>m>>maxn;
    for(int i=1;i<=n;i++)
    {
        string s;
        cin>>s;
        for(int j=0;j<s.length();j++)
        {
            qu[i].push(s[j]);//将每个传送带上的包装依次存入对应队列
        }
    }
    int x;
    while(cin>>x&&x!=-1)//当输入为-1时操作结束
    {
        if(x==0)//当0号按钮被按下时机械手将抓取筐顶部的一件物品放到流水线上
        {
            if(st.size())//如果筐内有包装
            {
                cout<<st.top();//将顶部取出
                st.pop();//将栈顶去除
            }
        }
        else if(x<=n&&st.size()!=maxn)//当是传送带的按钮时将传送带第一个的包装放入筐中
        {
            if(qu[x].size()!=0)//只有当传送带不为0时
            {
                int t=qu[x].front();
                qu[x].pop();
                st.push(t);
            }
        }
        else if(x<=n&&st.size()==maxn)//当筐已经满了，但仍然有某条轨道的按钮被按下时，系统应强制启动0号键
        {
            if(qu[x].size()!=0)
            {
                cout<<st.top();
                st.pop();//先从筐里抓出一件物品
                int t=qu[x].front();//再将对应轨道的物品推落
                qu[x].pop();
                st.push(t);
            }
        }
    }
    return 0;
}