UVa:11111 Generalized Matrioshkas

之前做数据结构遗留下来的一道简单题。

用栈来做，思路不难想。

定义一个结构体存尺寸和容积，读到负数则入栈，正数则岀栈（若栈空则false），如果栈顶元素尺寸与该正数巧合互为相反数（否则false），则将下一个栈顶元素的容积减去该正数，如果容积出现非正数则false。最后读完如果栈非空则false。

可能我没仔细读题，被很多细节给坑了。有不合法的也就是不是成对出现的那种数据。

 

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <stack>
using namespace std;
struct Toy
{
    int val,rest;
    void set(int a,int b)
    {
        val=a;
        rest=b;
    }
};
Toy d[10000];
int main()
{

    string str;
    while(getline(cin,str))
    {
        if(str.empty()) continue;
        int n=0,num=0;
        bool posi=true;
        for(int i=0;; ++i)
        {
            if(str[i]=='-') posi=false;
            else if(isdigit(str[i])) num=num*10+str[i]-'0';
            if((!isdigit(str[i])||i==str.size()-1)&&str[i]!='-')
            {
                if(num)
                {
                    if(!posi)
                        d[n++].set(-num,num);
                    else
                        d[n++].set(num,-num);
                    num=0;
                    posi=true;
                }
                if(i==str.size()-1) break;
            }
        }
        stack<Toy> sk;
        bool ok=true;
        for(int i=0; i<n&&ok; ++i)
        {
            if(d[i].val<0)
                sk.push(d[i]);
            else
            {
                if(!sk.empty())
                {
                    Toy gettop=sk.top();
                    if(-gettop.val==d[i].val)
                    {
                        sk.pop();
                        if(!sk.empty())
                        {
                            Toy tmp=sk.top();
                            sk.pop();
                            tmp.rest-=d[i].val;
                            if(tmp.rest<=0) ok=false;
                            sk.push(tmp);
                        }
                    }
                    else ok=false;
                }
                else ok=false;
            }
        }
        if(!sk.empty()) ok=false;
        if(ok) puts(":-) Matrioshka!");
        else puts(":-( Try again.");
    }
    return 0;
}