1247: 记忆数字
Time Limit: 5 Sec Memory Limit: 256 MBSubmit: 159 Solved: 17
[ Submit][ Status][ Web Board]
Description
wanghang最近研制出了一个机器人,这个机器人有一个跨时代的功能:记忆数字!
这个机器人接受三种指令
第一种指令,记忆一系列数字
第二种指令,询问一个数字是否在机器人的记忆中
第三种指令,让这个机器人忘记所有数字
Input
第一行一个正整数T<=10代表输入数据个数
第二行一个正整数N<=100000代表指令个数
然后N行,每行代表一个指令,指令格式如下:
1 X K代表记忆K个数字X,X+2,X+4 ... X+2K-2 (1<=K<=50)
2 Y 代表询问Y是否在机器人的记忆中
3 代表让机器人忘记所有数字
Output
对于每个第二种指令,输出Yes代表询问的数字在机器人的记忆中,反之输出No
Sample Input
151 1 52 22 332 2
Sample Output
NoYesNo
HINT
Source
考虑用hash表,插入和查找接近O(1),clear操作是O(n)。十万次查询,每次查询最多同时有五十万个节点需要清空,时间开销太大。如何解决clear的效率问题?
解决办法是lazy标记,仅当使用到时才清零。具体实现是用一个时间戳,再记录每个链表的版本号,一比较就知道是否需要更新。
感觉挺像可持久化编程的,其实不是。目的并不是要保留历史版本。而是通过标记需要清空的元素,不到使用的时候就不对他进行清空,于是我们的清零的时间全部分散到了插入和查询中。而对时间戳的更新只是O(1)的算法。插入和查询和原来的时间复杂度保持不变。因此这种Lazy标记的思路就是一个极好的优化。
注意,我们需要清零的是head数组,因此每一个链表里所有的元素都具有相同的时间戳,所以不应该对每个元素维护一个时间戳,这样对时间的优化是较小的
求hash值的时候,不要用abs。而对之and上0x7fffffff,直接舍掉符号位。这样会使hash值分散得多,因为正负值的hash差异很大。
提交时一直RE,原因是多组数据没有的hash表clear。导致内存池的指针一直往后移,超过了预期的最大值maxn*maxk。
#include <cstdio>
#include <set>
#include <map>
#include <cstring>
#include <queue>
#include <string>
const double eps = 1e-6;
const int INF = 0x3f3f3f3f;
const int maxnk = 5000005;
int getint()
{
int res = 0; char tmp; bool sgn = 1;
do tmp = getchar();
while (!isdigit(tmp) && tmp != '-');
if (tmp == '-')
{
sgn = 0;
tmp = getchar();
}
do res = (res << 1) + (res << 3) + tmp - '0';
while (isdigit(tmp = getchar()));
return sgn ? res : -res;
}
struct node
{
int v;
int next;
node(int vv,int _n):v(vv),next(_n){}
node(){}
};
const int Mod = 13131313;
struct HashTable
{
int top,version;
node pool[maxnk];
int head[Mod];
int history[Mod];
HashTable(){top = 0;memset(head,0,sizeof head);memset(history,0,sizeof history);version=0;}
bool find(int v)
{
int _hash = (v&0x7fffffff)%Mod;
if (history[_hash] < version)
{
head[_hash] = 0;
return false;
}
for (int i=head[_hash];i;i=pool[i].next)
{
if (pool[i].v == v)
return true;
}
return false;
}
void insert(int v)
{
int _hash = (v&0x7fffffff)%Mod;
bool ok = false;
if (history[_hash] < version)
{
head[_hash] = 0;
history[_hash] = version;
}
for (int i=head[_hash];i;i=pool[i].next)
{
if (pool[i].v == v)
{
ok = true;
return;
}
}
if (!ok)
{
top ++;
pool[top].v = v;
pool[top].next = head[_hash];
head[_hash] = top;
}
}
void clear()
{
top = 0;
//memset(head,0,sizeof head);
version ++;
}
};
HashTable table;
int main()
{
int t = getint();
while (t--)
{
table.clear();
int n = getint();
while (n--)
{
int o = getint();
if (o == 1)
{
int x = getint();
int k = getint();
for (int i=0;i<k;i++)
{
table.insert(x+i*2);
}
}
else if (o == 3)
{
table.clear();
}
else
{
int x = getint();
if (table.find(x))
printf("Yes\n");
else
printf("No\n");
}
}
}
return 0;
}