LibreOJ #6036.「雅礼集训 2017 Day4」编码 可持久化字典树+2-SAT

题意

有n个01串，某些字符串中的某一位可以任意填，问是否存在一种填法使得不存在某个串是另一个串的前缀。
$n\le500000$

分析

用可持久化字典树来优化2-SAT建图即可。
注意要把字符串按长度从大到小插入。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>

const int N=500005;

int n,sz,num,cnt,tim,top,last[N*10],dfn[N*10],low[N*10],stack[N*10],bel[N*10],id[N*4][2],tot,ch[N*4][2],rt,rk[N],len[N],now;
bool ins[N*10];
struct edge{int to,next;}e[N*40];
char str[N];
std::vector<int> a[N];

void addedge(int u,int v)
{
    if (!u||!v) return;
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}

void tarjan(int x)
{
    dfn[x]=low[x]=++tim;
    ins[x]=1;stack[++top]=x;
    for (int i=last[x];i;i=e[i].next)
        if (!dfn[e[i].to]) tarjan(e[i].to),low[x]=std::min(low[x],low[e[i].to]);
        else if (ins[e[i].to]) low[x]=std::min(low[x],dfn[e[i].to]);
    if (dfn[x]==low[x])
    {
        tot++;int y=0;
        while (y!=x)
        {
            y=stack[top];top--;
            bel[y]=tot;ins[y]=0;
        }
    }
}

void con(int x,int y)
{
    if (!y) return;
    addedge(id[y][0],id[x][0]);
    addedge(id[x][1],id[y][1]);
}

void insert(int &d,int nw,int u,int v)
{
    int p=d;d=++num;id[d][0]=++sz;id[d][1]=++sz;
    addedge(id[p][0],id[d][0]);addedge(id[d][1],id[p][1]);
    ch[d][0]=ch[p][0];ch[d][1]=ch[p][1];
    if (nw>len[now])
    {
        con(d,ch[d][0]);con(d,ch[d][1]);
        addedge(u,id[p][1]);addedge(id[p][0],v);
        addedge(u,id[d][0]);addedge(id[d][1],v);
        return;
    }
    insert(ch[d][a[now][nw]],nw+1,u,v);
    con(d,ch[d][0]);con(d,ch[d][1]);
}

bool cmp(int x,int y)
{
    return len[x]>len[y];
}

int main()
{
    scanf("%d",&n);
    sz=n*2;
    for (int i=1;i<=n;i++)
    {
        scanf("%s",str+1);
        len[i]=strlen(str+1);rk[i]=i;
        a[i].push_back(0);
        for (int j=1;j<=len[i];j++) a[i].push_back(str[j]=='0'?0:(str[j]=='1'?1:-1));
    }
    std::sort(rk+1,rk+n+1,cmp);
    for (int i=1;i<=n;i++)
    {
        now=rk[i];int pos=0;
        for (int j=1;j<=len[now];j++) if (a[now][j]==-1) {pos=j;break;}
        if (pos)
        {
            a[now][pos]=0;insert(rt,1,i,i+n);
            a[now][pos]=1;insert(rt,1,i+n,i);
        }
        else
        {
            insert(rt,1,i,i+n);
            addedge(i+n,i);
        }
    }
    for (int i=1;i<=sz;i++) if (!dfn[i]) tarjan(i);
    for (int i=1;i<=n;i++)
        if (bel[i]==bel[i+n]) {puts("NO");return 0;}
    puts("YES");
    return 0;
}