首页 > 代码库 > poj4093:倒排索引查询
poj4093:倒排索引查询
总时间限制: 1000ms 内存限制: 131072kB描述现在已经对一些文档求出了倒排索引,对于一些词得出了这些词在哪些文档中出现的列表。要求对于倒排索引实现一些简单的查询,即查询某些词同时出现,或者有些词出现有些词不出现的文档有哪些。输入第一行包含一个数N,1 <= N <= 100,表示倒排索引表的数目。接下来N行,每行第一个数ci,表示这个词出现在了多少个文档中。接下来跟着ci个数,表示出现在的文档编号,编号不一定有序。1 <= ci <= 1000,文档编号为32位整数。接下来一行包含一个数M,1 <= M <= 100,表示查询的数目。接下来M行每行N个数,每个数表示这个词要不要出现,1表示出现,-1表示不出现,0表示无所谓。数据保证每行至少出现一个1。输出共M行,每行对应一个查询。输出查询到的文档编号,按照编号升序输出。如果查不到任何文档,输出"NOT FOUND"。样例输入33 1 2 31 21 331 1 11 -1 01 -1 -1样例输出NOT FOUND1 31
这题直接用stl的set集合可以解决,直接贴代码,但是这题在刚开始的时候遇到了bug详见下一篇随笔。
#include <iostream>#include <stdio.h>#include <set>#include <algorithm>using namespace std;set<int> s[100];int n,m;int main(){ scanf("%d",&n); for(int i = 0; i < n; i++) { int k; scanf("%d",&k); for(int j = 0; j < k; j++) { int x; scanf("%d",&x); s[i].insert(x); } } scanf("%d",&m); for(int i = 0; i < m; i++) { set<int> a,b; bool flag = true; for(int j = 0; j < n; j++) { int f; scanf("%d",&f); if(f == 1&&flag) { flag = false; set<int>::iterator itr1 = s[j].begin(); set<int>::iterator itr2 = s[j].end(); while(itr1 != itr2) { a.insert(*itr1); itr1++; } } else if(f == 1&& !flag ) { set<int>::iterator itr1 = a.begin(); set<int>::iterator itr2 = a.end(); while(itr1 != itr2) { if(s[j].find(*itr1) == s[j].end()) //itr1 = a.erase(itr1); 这个地方有bug a.erase(itr1++); else itr1++; } } else if(f == -1) { set<int>::iterator itr1 = s[j].begin(); set<int>::iterator itr2 = s[j].end(); while(itr1 != itr2) { b.insert(*itr1); itr1++; } } } set<int>::iterator itr1 = b.begin(); set<int>::iterator itr2 = b.end(); while(itr1 != itr2) { a.erase(*itr1); itr1++; } if(a.empty()) printf("NOT FOUND\n"); else { set<int>::iterator itr = a.begin(); while(itr != a.end()) { printf("%d ",*itr); itr++; } printf("\n"); } } return 0;}
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。