UVa:11111 Generalized Matrioshkas

그동안 데이터 구조 에 남 겨 진 간단 한 문 제 를 풀 었 습 니 다.
창고 로 만 들 면 생각 하기 어렵 지 않다.
하나의 구조 체 저장 크기 와 용적 을 정의 하고 마이너스 까지 읽 으 면 스 택 에 들 어 갑 니 다. 정 수 는 23680 스 택 (스 택 이 비어 있 으 면 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;
}

좋은 웹페이지 즐겨찾기