图书管理系统（Borrowers,UVa230）

最新推荐文章于 2022-01-27 16:32:13 发布

三日看尽长安花

最新推荐文章于 2022-01-27 16:32:13 发布

阅读量193

点赞数

分类专栏： ACM

本文链接：https://blog.csdn.net/qq_42060170/article/details/104528062

版权

ACM 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

题目链接： 图书管理系统（UVa230）

C++代码：

#include <string>
#include <set>
#include <iostream>
#include <cassert>
#include <vector>
#include <map>
using namespace std;

struct Book{
    string title,author;
    Book(const string& t,const string&a):title(t),author(a){}
    bool operator<(const Book rhs) const{
        return author<rhs.author||(author==rhs.author&&title<rhs.title);
    }
};
vector<Book> books;
map<string,int> bookIndice;
struct indexComp{
    bool operator()(const int& lhs,const int& rhs)const{
        return books[lhs]<books[rhs];
    }
};

set<int,indexComp> shelf,lib;

void borrow(const string& t){
    assert(bookIndice.count(t));
    int idx=bookIndice[t];
    if(lib.count(idx)){
        lib.erase(idx);
    }else{
        assert(shelf.count(idx));
        shelf.erase(idx);
    }
}

void retBook(const string& t){
    assert(bookIndice.count(t));
    int idx=bookIndice[t];
    assert(!lib.count(idx));
    assert(!shelf.count(idx));
    shelf.insert(idx);
}

void shelve(){
    for(set<int>::iterator it=shelf.begin();it!=shelf.end();it++){
        int idx=*it;
        set<int>::iterator  pit=lib.insert(idx).first;
        if(pit==lib.begin())
            cout<<"Put "<<books[idx].title<<" first"<<endl;
        else{
            pit--;
            cout<<"Put "<<books[idx].title<<" after "<<books[*pit].title<<endl;
        }
    }
    shelf.clear();
    cout<<"END"<<endl;
}

int main(){
    string buf;
    while(true){
        getline(cin,buf);
        if(buf=="END") break;
        int spos=buf.find(" by ");
        assert(spos!=string::npos);
        string title=buf.substr(0,spos),author=buf.substr(spos+4);
        int idx=books.size();
        bookIndice[title]=idx;
        books.push_back(Book(title,author));
    }
    for(int i=0;i<books.size();i++)
        lib.insert(i);
    string cmd,title;
    while(true){
        getline(cin,buf);
        if(buf=="END") break;
        cmd=buf.substr(0,6);
        if(cmd[0]=='S') shelve();
        else{
            title=buf.substr(cmd.size()+1,buf.size());
            if(cmd[0]=='B') borrow(title);
            else{
                assert(cmd[0]=='R');
                retBook(title);
            }
        }
    }
    return 0;
}