hihoCoder.1513.小Hi的烦恼(bitset 五维偏序)

Posted 2021-01-12 sovietpower

题目链接

五维偏序，对每一维维护bitset，表示哪儿为1（比它大），然后5个bitset与起来就能得到答案了。
具体实现可以用5*n个bitset，按排名搞个前缀和。

复杂度(O(n^2/w))（本质是暴力的优化）。

//1284ms    565MB
#include <cstdio>
#include <cctype>
#include <bitset>
#include <algorithm>
#define gc() getchar()
const int N=30005;

int rk[N][5],pos[N];
std::bitset<N> bit[5][N];

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}

int main()
{
    int n=read();
    for(int i=1; i<=n; ++i)
        for(int j=0; j<5; ++j) rk[i][j]=read();
    for(int j=0; j<5; ++j)
    {
        for(int i=1; i<=n; ++i) pos[rk[i][j]]=i;
        for(int i=2; i<=n; ++i)
            bit[j][i]=bit[j][i-1], bit[j][i].set(pos[i-1]);
    }
    std::bitset<N> ans;
    for(int i=1; i<=n; ++i)
    {
        ans.set();
        for(int j=0; j<5; ++j) ans&=bit[j][rk[i][j]];
        printf("%d
",ans.count());
    }

    return 0;
}