【转载】[hrbust 2029] 二十世纪八十年代(状态压缩)


传送门:http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=2029

经典水题,给出一大堆东西然后问存不存在。给定的内存一共只有5M,所以不要试图存下所有字符串,这明显超内存。使用map或者字典树的话,内存恐怕也远超5M了吧。
考虑到最多只有5个字符,每个字符26种可能,所以可以把每一个字符串转化为一个整数。
26^5 = 11881376种状态。假设开一个bool数组,一个bool占一个字节,那么 11881376 / 1024 / 1024 = 11.33M
依然超内存。
状态肯定是要存的,一个状态只需要一个位就够了,一个字节的bool实在是浪费,那么接下来就是想办法将bool压缩到一个位了。
自己写也可以(文章最后放出),不过STL提供了个好东西,叫bitset,这样一个状态真的就只需要一个位了。
简单介绍一下bitset:

#include<bitset>

bitset<10> bt; // 开辟了有10个位的变量,即0000000000


bt.set(6); // 将第六位设为1,那么就变为0001000000

bt.set(1); // 将第1位设为1,那么就变为0001000010


bt.test(6); // 测试第六位是否为1,在这里返回1

bt.test(2); // 第二位是0,所以返回0

这样一来,问题就变得很简单了

#include<iostream>
#include<bitset>

using namespace std;

bitset<12000000> bs;

int main()
{
char str[6];
int n, m;
while(~scanf("%d\n", &n))
{
for(int i = 1; i <= n; i++)
{
scanf("%s", str);
int len = strlen(str);
int num = 0;
for(int i = 0; i < len; i++)
num = num * 26 + str[i] - 'a';

bs.set(num);
}
scanf("%d\n", &m);
for(int i = 1; i <= m; i++)
{
scanf("%s", str);
int len = strlen(str);
int num = 0;
for(int i = 0; i < len; i++)
num = num * 26 + str[i] - 'a';

if(bs.test(num))
printf("yes\n");
else
printf("no\n");
}
}
return 0;
}

那么不用bitset呢,当然也能做。开个数组就行了。数组的每一个位置能存32个位,
首先计算一下需要开多大的int数组:
26^5 / 32 = 371293,那么就开个38W大小的数组好了。
hash[0]存的是最开始的32个位,也就是0~31位
hash[1]存的是接下来的32个位,也就是32~63位
所以整数值映射到num数组就很简单了

x = Value / 32;

y = Value % 32;

也就是第X个数组位置的第y位。

赋值:hash[x] = hash[x] | (1 << y);

判断:if(hash[x] & (1 << y))

完整代码(实际运行时间比上面的bitset还要短,效果可真好!):

#include<iostream>
using namespace std;

int hash[380000];

int gethash(char *str)
{
int len = strlen(str);
int re = 0;
for(int i = 0; i < len; i++)
re = re * 26 + str[i] - 'a';
return re;
}

int main()
{
int n, m;
char str[6];
int num, x, y;
while(~scanf("%d\n", &n))
{
for(int i = 0; i < n; i++)
{
scanf("%s", str);
num = gethash(str);
x = num / 32;
y = num % 32;
hash[x] = hash[x] | (1 << y);
}
scanf("%d\n", &m);
for(int i = 0; i < m; i++)
{
scanf("%s", str);
num = gethash(str);
x = num / 32;
y = num % 32;
if(hash[x] & (1 << y))
printf("yes\n");
else
printf("no\n");
}
}
return 0;
}


你可能感兴趣的:(【转载】[hrbust 2029] 二十世纪八十年代(状态压缩))