一道题看bitset应用 --ZOJ 3642

题意：给n个文件，包括文件名和文件大小，然后给出k个关键词，查询包含该关键词的文件的大小总和。文件名为一些中括号括起的关键词的合集。

解法：可用bitset记录每一个关键词在哪些文件中出现，然后查询即可。

bitset用法如下：

bitset<SIZE> bs; bool is_set = bs.any();    //是否存在1

bool not_set = bs.none();  //是否全0

int cnt_one = bs.count(); bs[index] = 1; bs.flip(index); //反转第index位

bs[index].flip()   //反转第index位

bs.flip()          //反转所有

 bs.set();          //所有位赋为1

bs.reset();        //所有位赋为0

bs.set(index);     //第index位赋为1

bs.reset(index)    //第index位赋为0



string bitval("1010"); bitset<32> bs2(bitval);   //初始化

 

用到了指针和strchr函数，strchr(str,ch)函数用来找出str指向的字符串中第一次出现字符ch的位置。

代码：

#include <iostream>

#include <cstdio>

#include <cstring>

#include <cmath>

#include <algorithm>

#include <string>

#include <map>

#include <bitset>

#include <cstdlib>

#define ll long long

using namespace std;

#define N 1207



bitset<N> mask;

ll siz[N];

char ss[N];

char *a,*b;

map<string,bitset<N> > mp;



int main()

{

    int n,k,i,j;

    while(scanf("%d",&n)!=EOF)

    {

        mp.clear();

        for(i=1;i<=n;i++)

        {

            scanf("%s %lld",ss,&siz[i]);

            a = ss;

            while((a = strchr(a,'[')) != NULL)

            {

                b = strchr(a,']');

                string tmp = string(a+1,b);

                mp[tmp].set(i-1);

                a = b;

            }

        }

        scanf("%d",&k);

        while(k--)

        {

            scanf("%s",ss);

            for(j=0;j<n;j++)

                mask.set(j);

            a = ss;

            while((a = strchr(a,'[')) != NULL)

            {

                b = strchr(a,']');

                string tmp = string(a+1,b);

                if(!mp.count(tmp))

                {

                    mask.reset();

                    break;

                }

                else

                    mask &= mp.find(tmp)->second;

                a = b;

            }

            ll res = 0;

            for(j=0;j<n;j++)

            {

                if(mask[j])

                    res += siz[j+1];

            }

            printf("%lld\n",res);

        }

    }

    return 0;

}

View Code