题目大意:
给出若干本书及其作者,输入保证书名和作者长度不超过80个字符,而且书名中不会出现双引号,以END结束。接下来有若干个操作,分为三种:
1、BORROW操作:后面跟一个书名,表示借走这本书
2、RETURN操作:后面跟一个书名,表示归还这本书,但是这本书不会立即归还到书架上
3、SHELVE操作:后面不会跟任何东西,表示把所有归还的数放到书架上,书架上按照作者字典序升序排序,作者相同时,按照书名字典序升序排序,放到书架上,并输出 Put title1 after title2 或者 Put title first,取决于当前这本书是否被放到第一个位置,这若干个操作同样以END结尾
思路:
对每一本书做一个编号,用两个set维护书架和归还的书籍,在归还书籍时,顺便把这个编号插入到书架中,这样在输出时只要判断这本书是不是第一个就可以了
个人总结:
1、set会自动实现排序,所以自定义数据类型的时候需要重载小于号(本题没用到这个)
2、假设a是一个string的对象,那么可以根据a.c_str()获取一个字符数组,这样就可以在printf里输出了
3、在判断END和三种操作的时候,可以只判断字符串的第一个字符,效率也会高一点
代码:
#include <set>
#include <map>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
//用一个结构体存储一本书
struct book
{
string t;
string o;
//因为需要先对所有书籍排序,所以要重载<运算符
bool operator < (const book & b){
return o < b.o || (o == b.o and t < b.t);
}
};
string s , t;
vector < book > v;
int main(void)
{
while(getline(cin , s) && s[0] != 'E')
{
int i = s.find('"' , 1);//找到第二个"的位置
t = s.substr(i + 5);//作者
s = s.substr(0 , i + 1);//书名,我这里书名是直接挂着两边的双引号的
book newBook;
newBook.t = s;
newBook.o = t;
v.push_back(newBook);//插入到vector中
}
sort(v.begin() , v.end());//排序
map < string , int > mp;//映射,int代表每本书的编号
set< int > a , b;//a是书架,b是归还的书籍
for(int i = 0 ; i < (int)v.size() ; ++ i)
{
mp[v[i].t] = i;//对每一本书做编号
a.insert(i);//插入到书架上
}
while(getline(cin , s) && s[0] != 'E')
{
if(s[0] == 'B')//借书操作
{
s = s.substr(7);//书名,也是挂着两边的双引号的
a.erase(mp[s]);//直接将书架中的这本书删除
}
else if(s[0] == 'R')//还书操作
{
s = s.substr(7);//书名,也是挂着两边的双引号的
b.insert(mp[s]);//插入到归还书籍中
a.insert(mp[s]);//同时插入到书架上
}
else//归还书架的操作
{
for(auto i : b)//遍历
{
auto idx = a.find(i);//找到这本书放到书架上之后的位置
if(idx == a.begin())//如果他是第一个
printf("Put %s first\n" , v[i].t.c_str());
else//如果不是第一个
printf("Put %s after %s\n" , v[i].t.c_str() , v[*(--idx)].t.c_str());
a.insert(i);
}
b.clear();
puts("END");//别忘了输出END
}
}
return 0;
}