1022.Digital Library (30)...to be continued...

1022.Digital Library (30)…to be continued…

pat-al-1022

2017-02-20

• 关于map的用法：如果是一一对应关系，直接令mp[index] = content就可以；如果是一对多，就要用到mp[index].insert
• getline的用法：getline(cin, input)。别忘了如果前一个输入是scanf，要读掉换行
• map和set的方法中都有find
• 判断一个查询内容是否存在可以用：if(mp.find(queryContent) == mp.end())，或者if(mp.count(queryContent) == 0)，或者if(mp[queryContent].size() == 0)

/**
 * pat-al-1022
 * 2017-02-20
 * Cpp version
 * Author: fengLian_s
 */
#include<stdio.h>
#include<iostream>
#include<map>
#include<set>
#include<string>
using namespace std;
map<string, set<int> > mpTitle, mpAuthor, mpKey, mpPub, mpYear;
void query(map<string, set<int> >& mp, string& queryContent)
{
  if(mp[queryContent].size() == 0)
  {
    printf("Not Found\n");
  }
  else
  {
    for(set<int>::iterator it = mp[queryContent].begin();it != mp[queryContent].end();it++)
    {
      printf("%07d\n", *it);
    }
  }
}
int main()
{
  freopen("in.txt", "r", stdin);
  int n, m, id, type;
  string title, author, key, pub, year;
  scanf("%d", &n);
  for(int i = 0;i < n;i++)
  {
    scanf("%d\n", &id);
    getline(cin, title);
    //cout << "title = " << title << endl;
    mpTitle[title].insert(id);
    getline(cin, author);
    //cout << "author = " << author << endl;
    mpAuthor[author].insert(id);
    while(cin >> key)
    {
      char c = getchar();
      mpKey[key].insert(id);
      //cout << "key = " << key << endl;
      if(c == '\n')
        break;
    }
    getline(cin, pub);
    //cout << "pub = " << pub << endl;
    mpPub[pub].insert(id);
    getline(cin, year);
    //cout << "year = " << year << endl;
    mpYear[year].insert(id);
  }
  scanf("%d", &m);
  string queryContent;
  for(int i = 0;i < m;i++)
  {
    scanf("%d: ", &type);
    getline(cin, queryContent);
    printf("%d: ", type);
    cout << queryContent << endl;
    if(type == 1)
      query(mpTitle, queryContent);
    else if(type == 2)
      query(mpAuthor, queryContent);
    else if(type == 3)
      query(mpKey, queryContent);
    else if(type == 4)
      query(mpPub, queryContent);
    else
      query(mpYear, queryContent);
  }
  return 0;
}

-TBC-