【JZOJ5328】【NOIP2017提高组模拟】世界线（STL）

Description

Solution

这题刚看的时候就知道是用bitset来做，但是比赛的时候并不知道要怎么打，所以就只用了set来打。
比赛之后学了一下bitset发现bitset其实就是帮你把二进制状压了一下。时间和空间都是除以32的。
然后拓扑排序一下，倒着把点的集合合并到前面去。
但是直接这样做bitset会爆空间，所以我们可以考虑每次只存[l,r]的点，这样就可以用时间来换空间。
复杂度n^2/32

Code

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<bitset>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
#define rep(i,a) for(i=first[a];i;i=next[i])
using namespace std;
const int maxn=1e5+7;
typedef long long ll;
int i,j,k,l,t,n,m,x,head,tail,y,r;
int first[maxn*2],last[maxn*2],next[maxn*2],num,in[maxn],out[maxn];
int size[maxn];
int d[maxn];
ll ans;
bitset<1000>s[maxn];
void add(int x,int y){
    last[++num]=y,next[num]=first[x],first[x]=num;
}
int main(){
    freopen("worldline.in","r",stdin);
    freopen("worldline.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,m){
        scanf("%d%d",&k,&l),add(k,l),in[l]++;
    }
    fo(i,1,n)if(!in[i])d[++tail]=i;
    while(head<tail){
        rep(i,d[++head]){
            in[last[i]]--;
            if(!in[last[i]])d[++tail]=last[i];
        }
    }
    for(l=1;l<=n;l+=1000){
        r=min(n,l+999);
        fod(i,n,1){
            x=d[i];s[x].reset();
            rep(j,x){
                s[x]|=s[last[j]];
                if(last[j]>=l&&last[j]<=r)s[x].set(last[j]-l);
            }
            ans+=s[x].count();
        }
    }
    printf("%d\n",ans-m);
}