題目大意:
給出若干本書及其作者,輸入保證書名和作者長度不超過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;
}