LOJ#3101. 「JSOI2019」精准预测
设0是生，1是死，按2-sat连边那么第一种情况是((t,x,1) ightarrow (t + 1,y,1))，((t + 1,y, 0) ightarrow (t,x,0))
第二种情况是((t,x,0) ightarrow (t,y,1)),((t,y,0) ightarrow(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('
')
#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();
}