暑假集训 || bitset

bitset是一个存储0和1的数组

可以快速的把两个bitset的每一位对应做与或啥的

在可以用01串表示某个状态的时候可以应用到它

就是有两个集合，求它们的交集

bitset <> a,b,c;
a[] = a[] = a[] = ;
b[] = b[] = b[] = b[] = ;
c = a & b;
cout << c << endl;
cout << "size = " << c.count() << endl;

HihoCoder 1513

题意：给出n个学生，他们5门课的排名，问对每个学生，每科都有多少人排在它前面

思路：n是30000，n^2在bitset下可以过的，因为每个学生的排名不同，所以bs[i][j]保存第 i 门课，排名为 j 的学生，有哪些人排名比他高

这里要注意储存的状态的选择，因为输出的时候每个学生我们只知道它该门课它的排名，而对于这门课来说，不同排名的人的bs数组肯定是不同的，所以 j 的状态选择即为排名为 j

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <bitset>
using namespace std;
const int SZ = ;
const int INF = ;
int mp[SZ][];
int a[SZ], b[SZ], c[SZ], d[SZ], e[SZ];
bitset<SZ> bs[][SZ], ans;
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = ; i <= n; i++)
    {
        scanf("%d %d %d %d %d", &a[i], &b[i], &c[i], &d[i], &e[i]);
        mp[a[i]][] = i, mp[b[i]][] = i, mp[c[i]][] = i, mp[d[i]][] = i, mp[e[i]][] = i;
    }
    for(int i = ; i < ; i++)
        for(int j = ; j <= n; j++)
        {
            bs[i][j] = bs[i][j-];
            bs[i][j][mp[j-][i]] = ;
            //cout << i <<" "<< j <<" "<< bs[i][j] << endl;
        }
    for(int i = ; i <= n; i++)
    {
        ans = bs[][a[i]] & bs[][b[i]] & bs[][c[i]] & bs[][d[i]] & bs[][e[i]];
        printf("%d\n", ans.count());
    }
    return ;
}

Gym 100342J

题意：给n个城市，有些城市间有有向边，求三元环的个数（三元环：A->B->C->A）

思路：想不到bitset哇。。。把每个城市能到达的城市，和能被到达的用bitset表示出来

循环n个城市（A），对每个A找到它能到达的B（用vector存）

B能到达的城市，和A能被到达的城市，这两个集合的交集就是C，&一下然后count1的个数即可

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <bitset>
#include <vector>
using namespace std;
typedef long long LL;
const int SZ = ;
const int INF = ;
int vis[SZ][SZ];
char s[SZ][SZ];
bitset<SZ> bs[][SZ];
vector<int> v[SZ];
int main()
{
    freopen("triatrip.in", "r", stdin);
    freopen("triatrip.out", "w", stdout);
    int n;
    scanf("%d", &n);
    for(int i = ; i < n; i++)
        scanf(" %s", s[i]);
    for(int i = ; i < n; i++)
        for(int j = ; j < n; j++)
            if(s[i][j] == '+')
            {
                bs[][i][j] = ;
                bs[][j][i] = ;
                v[i].push_back(j);
            }
    LL cnt = ;
    for(int i = ; i < n; i++)
        for(int j = ; j < v[i].size(); j++)
            cnt += (LL)((bs[][v[i][j]] & bs[][i]).count());
    printf("%lld\n", cnt/);
    return ;
}