LOJ #6036.「雅礼集训 2017 Day4」编码 Trie树上2-sat

记得之前做过几道2-sat裸体,以及几道2-sat前缀优化建图,这道题使用了前缀树上前缀树优化建图.
我们暴力建图肯定是n^2级别的,那么我们要是想让边数少点,就得使用一些骚操作.
我们观察我们的限制条件,不就是选了一个点,那么这个点的前缀都不能选吗(选了一个点,以他为前缀的的点也不能选,这个限制条件可以通过前面那个限制条件体现出来,所以说观察到问题本质是一样的,可以简化我们的问题).那么我们就可以在Trie上建图,使得选择一个点,那么他的前缀点都必须不能选,就可以了.但是对于一个点上有多个点的情况,我们要特殊处理,我的处理方法是,把那些点拽出来作为树点.
最后说一下,要注意2-sat建图一定要连逆否命题边.
(感觉自己2-sat好虚啊……)
语言笔记:
exit(0),只要你在程序中使用,无论在哪里,直接正常退出,具体情况见http://blog.csdn.net/u010046690/article/details/47105665

#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
const int N=;
struct V{
  int to,next;
}c[N<<];
int head[N<<],t;
inline void add(int x,int y){
  c[++t].to=y,c[t].next=head[x],head[x]=t;
}
char s[N],*begin[N];
int n,len[N],m;
int pos[N],temp[N];
int dfn[N<<],low[N<<],belong[N<<],stack[N<<],top,num,Ti;
bool in[N<<];
inline void Tarjan(int x){
  dfn[x]=low[x]=++Ti;
  stack[++top]=x;
  in[x]=true;
  register int i;
  for(i=head[x];i;i=c[i].next)
    if(!dfn[c[i].to]){
      Tarjan(c[i].to);
      low[x]=std::min(low[x],low[c[i].to]);
    }else if(in[c[i].to])
      low[x]=std::min(low[x],dfn[c[i].to]);
  if(low[x]==dfn[x]){
    ++num;
    register int j;
    do{
      j=stack[top--];
      in[j]=false;
      belong[j]=num;
    }while(j!=x);
  }
}
#define choose(a,b) (((a)<<1)-(b))
struct Trie{
  Trie *ch[];
  std::vector<int> mem;
  Trie(){ch[]=ch[]=NULL,mem.clear();}
  inline void* operator new(size_t);
}*root,*C,*mempool;
inline void* Trie::operator new(size_t){
  if(C==mempool){
    C=new Trie[(<<)+];
    mempool=C+(<<)+;
  }
  return C++;
}
inline void build(Trie *p,int id){
  register int i,x,a,b,size=p->mem.size();
  if(!size)return;
  for(i=;i<size;++i){
    x=p->mem[i];
    if((pos[x]==-&&pos[id]==-)||(pos[x]==-&&pos[id]>=len[x])){
      puts("NO");
      exit();
    }
    if(pos[x]==pos[id]){
      add(choose(x,),choose(id,));
      add(choose(x,),choose(id,));
      add(choose(id,),choose(x,));
      add(choose(id,),choose(x,));
    }else{
      if(pos[id]==-||pos[id]>=len[x]){
        if(begin[id][pos[x]]==''){
          add(choose(x,),choose(x,));
        }else{
          add(choose(x,),choose(x,));
        }
      }else{
        if(pos[x]==-){
          if(begin[x][pos[id]]==''){
            add(choose(id,),choose(id,));
          }else{
            add(choose(id,),choose(id,));
          }
        }else{
          a=begin[id][pos[x]]=='';
          b=begin[x][pos[id]]=='';
          add(choose(x,a),choose(id,b^));
          add(choose(id,b),choose(x,a^));
        }
      }
    }
  }
}
inline void insert(int id){
  register Trie *p=root,*w;
  register int i,j,l=len[id];
  for(i=;i<l;++i){
    if(begin[id][i]=='?'){
      if(!p->ch[])
        p->ch[]=new Trie();
      w=p->ch[];
      build(w,id);
      for(j=i+;j<l;++j){
        if(!w->ch[begin[id][j]-''])
          w->ch[begin[id][j]-'']=new Trie();
        w=w->ch[begin[id][j]-''];
        build(w,id);
      }
      w->mem.push_back(id);
      if(!p->ch[])
        p->ch[]=new Trie();
      w=p->ch[];
      build(w,id);
      for(j=i+;j<l;++j){
        if(!w->ch[begin[id][j]-''])
          w->ch[begin[id][j]-'']=new Trie();
        w=w->ch[begin[id][j]-''];
        build(w,id);
      }
      w->mem.push_back(id);
      return;
    }
    if(!p->ch[begin[id][i]-''])
      p->ch[begin[id][i]-'']=new Trie();
    p=p->ch[begin[id][i]-''];
    build(p,id);
  }
  p->mem.push_back(id);
}
inline bool comp(int x,int y){
  return len[x]<len[y];
}
inline void Init(){
  root=new Trie();
  scanf("%d",&n);
  register int i,j;
  for(i=;i<=n;++i){
    pos[i]=-;
    begin[i]=s+m;
    scanf("%s",s+m);
    len[i]=strlen(s+m);
    m+=len[i];
    temp[i]=i;
    for(j=;j<len[i];++j)
      if(begin[i][j]=='?'){
        pos[i]=j;
        break;
      }
  }
  std::sort(temp+,temp+(n+),comp);
  for(i=;i<=n;++i){
    insert(temp[i]);
  }
}
inline bool check(){
  register int i;
  for(i=;i<=(n<<);++i)
    if(!dfn[i])
      Tarjan(i);
  for(i=;i<=n;++i)
    if(belong[choose(i,)]==belong[choose(i,)])
      return false;
  return true;
}
int main(){
  //freopen("rio.in","r",stdin);
  Init();
  puts(check()?"YES":"NO");
  return ;
}

88分暴力

#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define R register
const int N=;
struct V{int to,next;}c[N<<];
int head[N*],t;
inline void add(int x,int y){
  c[++t].to=y,c[t].next=head[x],head[x]=t;
}
char s[N];
int n,cnt;
int dfn[N*],low[N*],belong[N*],stack[N*],top,num,Ti;
bool in[N*];
inline void Tarjan(int x){
  dfn[x]=low[x]=++Ti;
  stack[++top]=x;
  in[x]=true;
  R int i;
  for(i=head[x];i;i=c[i].next)
    if(!dfn[c[i].to]){
      Tarjan(c[i].to);
      low[x]=std::min(low[x],low[c[i].to]);
    }else if(in[c[i].to])
      low[x]=std::min(low[x],dfn[c[i].to]);
  if(low[x]==dfn[x]){
    ++num;
    do{
      i=stack[top--];
      in[i]=false;
      belong[i]=num;
    }while(i!=x);
  }
}
#define H(a,b) (((a)<<1)-(b))
struct Trie{
  Trie *ch[];
  std::vector<int> mem;
  Trie(){ch[]=ch[]=NULL,mem.clear();}
  inline void* operator new(size_t);
}*root,*C,*mempool;
inline void* Trie::operator new(size_t){
  if(C==mempool){
    C=new Trie[(<<)+];
    mempool=C+(<<)+;
  }
  return C++;
}
inline void dfs(Trie *p,int fa){
  if(!p)return;
  R int i,j,x,last=fa,size=p->mem.size();
  for(i=;i<size;++i){
    x=p->mem[i];
    ++cnt;
    if(x>)j=;
    else x=-x,j=;
    add(H(x,j),H(cnt,));
    add(H(cnt,),H(x,j^));
    add(H(x,j),H(last,));
    add(H(last,),H(x,j^));
    add(H(cnt,),H(last,));
    add(H(last,),H(cnt,));
    last=cnt;
  }
  ++cnt;
  if(last){
    add(H(cnt,),H(last,));
    add(H(last,),H(cnt,));
  }
  last=cnt;
  dfs(p->ch[],last);
  dfs(p->ch[],last);
}
inline void insert(int id){
  scanf("%s",s);
  R Trie *p=root,*w;
  R int i,j,l=strlen(s);
  for(i=;i<l;++i)s[i]-='';
  for(i=;i<l;++i){
    if(s[i]>){
      w=(!p->ch[])?p->ch[]=new Trie():p->ch[];
      for(j=i+;j<l;++j)
        w=(!w->ch[s[j]])?w->ch[s[j]]=new Trie():w->ch[s[j]];
      w->mem.push_back(-id);
      w=(!p->ch[])?p->ch[]=new Trie():p->ch[];
      for(j=i+;j<l;++j)
        w=(!w->ch[s[j]])?w->ch[s[j]]=new Trie():w->ch[s[j]];
      w->mem.push_back(id);
      return;
    }
    p=(!p->ch[s[i]])?p->ch[s[i]]=new Trie():p->ch[s[i]];
  }
  add(H(id,),H(id,));
  p->mem.push_back(id);
}
inline bool check(){
  R int i;
  for(i=;i<=(cnt<<);++i)
    if(!dfn[i])Tarjan(i);
  for(i=;i<=cnt;++i)
    if(belong[H(i,)]==belong[H(i,)])
      return false;
  return true;
}
int main(){
  root=new Trie();
  scanf("%d",&n),cnt=n;
  R int i;
  for(i=;i<=n;++i)insert(i);
  dfs(root,);
  puts(check()?"YES":"NO");
  return ;
}

100分正解