【LOJ】#3101. 「JSOI2019」精準預測

LOJ#3101. 「JSOI2019」精準預測

設0是生，1是死，按2-sat連邊那麼第一種狀況是\((t,x,1) \rightarrow (t + 1,y,1)\)，\((t + 1,y, 0) \rightarrow (t,x,0)\)

第二種狀況是\((t,x,0) \rightarrow (t,y,1)\),\((t,y,0) \rightarrow(t,x,1)\)

而後\((t,x,0)\)往\((t - 1,x,0)\)連邊，\((t,x,1)\)往\((t + 1,x,1)\)連邊

發現咱們顯然能夠忽略出度只有一個的點，最後只剩下\(2m + 2n\)個點

而後這是一張DAG，由於生的圖是一張DAG，死的圖是一張DAG，剩下的只有從生連到死的邊

因而咱們要求從\((T + 1,x,0)\)走到的\((T + 1,y,1)\)的個數，而後減去

同時咱們還要給每一個點減去必定會死的人

用bitset優化一下便可。。。

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define space putchar(' ')
#define enter putchar('\n')
#define eps 1e-10
#define ba 47
#define MAXN 50005
//#define ivorysi
using namespace std;
typedef long long int64;
typedef unsigned int u32;
typedef double db;
template<class T>
void read(T &res) {
    res = 0;T f = 1;char c = getchar();
    while(c < '0' || c > '9') {
    if(c == '-') f = -1;
    c = getchar();
    }
    while(c >= '0' && c <= '9') {
    res = res * 10 +c - '0';
    c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {x = -x;putchar('-');}
    if(x >= 10) {
    out(x / 10);
    }
    putchar('0' + x % 10);
}
struct node {
    int to,next;
}E[2000005];
struct predict {
    int ty,t,x,y;
}pre[100005];
int T,N,M,Ncnt;
vector<int> tim[2][MAXN],pos[2][MAXN];
bitset<50005> reach[100005],all,res[50005];
int head[300005],sumE,deg[300005],rec[300005],dead[300005],live[300005];
vector<int> que,used;
int pu[300005],ans[50005];
void add(int u,int v) {
    if(!v || !u) return; 
    E[++sumE].to = v;
    E[sumE].next = head[u];
    ++deg[v];++rec[v];
    head[u] = sumE;
}
int getid(int k,int x,int t) {
    if(k == 0) {
    int tar = upper_bound(tim[k][x].begin(),tim[k][x].end(),t) - tim[k][x].begin() - 1;
    if(tar < 0) return 0;
    return pos[k][x][tar];
    }
    else {
    int tar = lower_bound(tim[k][x].begin(),tim[k][x].end(),t) - tim[k][x].begin();
    return pos[k][x][tar];
    }
}
void Init() {
    read(T);read(N);read(M);
    Ncnt = 0;
    for(int i = 1 ; i <= N ; ++i) {tim[1][i].pb(T + 1);tim[0][i].pb(T + 1);}
    for(int i = 1 ; i <= M ; ++i) {
    read(pre[i].ty);read(pre[i].t);read(pre[i].x);read(pre[i].y);
    if(pre[i].ty == 0) {
        tim[1][pre[i].x].pb(pre[i].t);
        tim[0][pre[i].y].pb(pre[i].t + 1);
    }
    else {
        
        tim[0][pre[i].x].pb(pre[i].t);
        tim[0][pre[i].y].pb(pre[i].t);
    }
    }
    for(int i = 1 ; i <= N ; ++i) {
    for(int k = 0 ; k < 2 ; ++k) {
        sort(tim[k][i].begin(),tim[k][i].end());
        tim[k][i].erase(unique(tim[k][i].begin(),tim[k][i].end()),tim[k][i].end());
        for(int j = 0 ; j < tim[k][i].size() ; ++j) {
        pos[k][i].pb(++Ncnt);
        if(k == 1) dead[Ncnt] = i;
        if(k == 0 && j != 0) add(pos[k][i][j],pos[k][i][j - 1]);
        if(k == 1 && j != 0) add(pos[k][i][j - 1],pos[k][i][j]); 
        }
    }
    live[pos[0][i].back()] = i;
    }
    for(int i = 1 ; i <= M ; ++i) {
    if(pre[i].ty == 0) {
        add(getid(1,pre[i].x,pre[i].t),getid(1,pre[i].y,pre[i].t + 1));
        add(getid(0,pre[i].y,pre[i].t + 1),getid(0,pre[i].x,pre[i].t));
    }
    else {
        add(getid(0,pre[i].x,pre[i].t),getid(1,pre[i].y,pre[i].t));
        add(getid(0,pre[i].y,pre[i].t),getid(1,pre[i].x,pre[i].t));
    }
    }
    
    
}
void Solve() {
    queue<int> Q;
    for(int i = 1 ; i <= Ncnt ; ++i) {
    if(!deg[i]) Q.push(i);
    }
    while(!Q.empty()) {
    int u = Q.front();Q.pop();que.pb(u);
    for(int i = head[u] ; i ; i = E[i].next) {
        int v = E[i].to;
        if(!--deg[v]) Q.push(v);
    }
    }
    for(int i = 0 ; i <= 100000 ; ++i) used.pb(i);
    
    all.reset();
    for(int j = que.size() - 1 ; j >= 0 ; --j) {
    int u = que[j];pu[u] = used.back();used.pop_back();
    reach[pu[u]].reset();
    if(dead[u]) reach[pu[u]][dead[u]] = 1;
    //cerr << "???" << endl;
    for(int i = head[u] ; i ; i = E[i].next) {
        int v = E[i].to;
        reach[pu[u]] |= reach[pu[v]];
        if(!--rec[v]) {
        used.pb(pu[v]);
        }
    }
    //cerr << "????" << endl;
    if(live[u]) {
        if(!reach[pu[u]][live[u]]) res[live[u]] = reach[pu[u]];
        else all[live[u]] = 1;
    }
    if(!rec[u]) used.pb(pu[u]);
    //cerr << "?????" << endl;
    }
    for(int i = 1 ; i <= N ; ++i) {
    if(!all[i]) {
        ans[i] = N - 1 - (res[i] | all).count();
    }
    out(ans[i]);space;
    }
    enter;
}
int main(){
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Init();
    Solve();
}