[2-SAT]NOI2017 .day2 T1 游戏

一看题目 3-sat???
然而发现为x的地图最多有8个,那么可以枚举每个x是什么,然后2-sat,复杂度是 38(n+m) ,再想一想,其实可以枚举每个x是a,或者用A型车,再用2-sat,复杂度就是 28(n+m) 了,但是建图太鬼畜,程序一大半在建图,最后还是被卡成68分

#include 
#include 
#include 
#include 

using namespace std;

const int N=500010;

int n,m;
int a[N],b[N],Q[N],l,r;
char s[N];
vector<int> xp,back;
struct edge{
  int a,b;
  char x,y;
}e[N];
vector inn[N],outt[N];

int G[N],bel[N],stk[N],vis[N],low[N],dfn[N],cnt,top,tms,g;
struct Edge{
  int t,nx;
}E[N];

inline void add(int x,int y){
  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;
}

void tarjan(int x){
  vis[x]=1;
  dfn[x]=low[x]=++tms;
  stk[++top]=x;
  for(int i=G[x];i;i=E[i].nx){
    if(!vis[E[i].t]) tarjan(E[i].t);
    if(vis[E[i].t]==1) low[x]=min(low[x],low[E[i].t]);
  }
  if(low[x]==dfn[x]){
    ++g; int k;
    do{
      k=stk[top--];
      bel[k]=g;
      vis[k]=2;
    }while(top && k!=x);
  }
}

int iG[N],opp[N],d[N],icnt;
struct iedge{
  int t,nx;
}ie[N];

inline void Add(int x,int y){
  ie[++icnt].t=y; ie[cnt].nx=iG[x]; iG[x]=icnt;
}

inline bool Back(bool x){
  for(int i=0;i'x';
  return x;
}

inline bool judge(){
  l=r=0; back.clear();
  for(int i=1;i<=n;i++) a[i]=0;
  for(int i=0;iif(b[xp[i]]) a[xp[i]]=1;
  for(int i=1;i<=m;i++)
    if(!a[e[i].a] && !a[e[i].b] && e[i].y-'A'==s[e[i].b]-'a' && e[i].x-'A'!=s[e[i].a]-'a'){
      if(s[e[i].a]=='x'){
    s[e[i].a]=e[i].x-'A'+'a'; back.push_back(e[i].a);
    continue;   
      }
      a[e[i].a]=6-(e[i].x-'A'+1)-(s[e[i].a]-'a'+1); 
    }

  for(int i=1;i<=n;i++)
    if(a[i]) Q[r++]=i;
  while(lint x=Q[l++];
    for(int i=0;iif(e.x-'A'+1!=a[x]) continue;
      if(a[e.b] && a[e.b]!=e.y-'A'+1) return Back(false);
      if(a[e.b]) continue;
      a[e.b]=e.y-'A'+1;
      Q[r++]=e.b;
    }
  }
  l=r=0;
  for(int i=1;i<=n;i++)
    if(a[i]) Q[r++]=i;
  while(lint x=Q[l++];
    for(int i=0;iif(e.y-'A'+1==a[x]) continue;
      if(a[e.a] && a[e.a]==e.x-'A'+1) return Back(false);
      if(a[e.a]) continue;
      if(e.x-'A'==s[e.a]-'a') continue;
      if(s[e.a]=='x'){
    s[e.a]=e.x-'A'+'a',back.push_back(e.a);
    continue;
      }
      a[e.a]=6-(e.x-'A'+1)-(s[e.a]-'a'+1);
      Q[r++]=e.a;
    }
  }
  for(int i=1;i<=n;i++) if(a[i]+'a'-1==s[i]) return Back(false);
  for(int i=1;i<=n;i++) G[i]=0; cnt=0;
  for(int i=1;i<=m;i++)
    if(!a[e[i].a] && !a[e[i].b]){
      if(e[i].x-'A'==s[e[i].a]-'a') continue;
      int x=e[i].x-'A'+1,y=6-(e[i].x-'A'+1)-(s[e[i].a]-'a'+1),u=e[i].y-'A'+1,v=6-(e[i].y-'A'+1)-(s[e[i].b]-'a'+1);
      if(x>y) swap(x,y); if(u>v) swap(u,v);
      if(x==e[i].x-'A'+1 && e[i].y-'A'+1==u) add(e[i].a<<1,e[i].b<<1),add(e[i].b<<1|1,e[i].a<<1|1);
      if(x==e[i].x-'A'+1 && e[i].y-'A'+1==v) add(e[i].a<<1,e[i].b<<1|1),add(e[i].b<<1,e[i].a<<1|1);
      if(y==e[i].x-'A'+1 && e[i].y-'A'+1==u) add(e[i].a<<1|1,e[i].b<<1),add(e[i].b<<1|1,e[i].a<<1);
      if(y==e[i].x-'A'+1 && e[i].y-'A'+1==v) add(e[i].a<<1|1,e[i].b<<1|1),add(e[i].b<<1,e[i].a<<1);
    }
  for(int i=1;i<=n;i++) vis[i]=bel[i]=0; g=0;
  for(int i=2;i<=(n<<1|1);i++)
    if(!vis[i]) tarjan(i),top=tms=0;
  for(int i=1;i<=n;i++)
    if(bel[i<<1]==bel[i<<1|1]) return Back(false);
  for(int i=1;i<=g;i++) iG[i]=opp[i]=0; icnt=0;
  for(int k=2;k<=(n<<1|1);k++){
    for(int i=G[k];i;i=E[i].nx)
      if(bel[E[i].t]!=bel[k])
    Add(bel[E[i].t],bel[k]);
    opp[bel[k]]=bel[k^1];           
  }
  l=r=0;
  for(int i=1;i<=g;i++) vis[i]=0;
  for(int i=1;i<=g;i++)
    if(!d[i]) Q[r++]=i;
  while(lint x=Q[l++];
    if(!vis[x]) vis[x]=1,vis[opp[x]]=2;
    for(int i=iG[x];i;i=ie[i].nx){
      if(!(--d[ie[i].t])) Q[r++]=ie[i].t;
    }
  }
  for(int i=1;i<=n;i++){
    if(a[i]){
      putchar(a[i]+'A'-1); continue;
    }
    int u,v;
    if(s[i]=='a') u=2,v=3;
    if(s[i]=='b') u=1,v=3;
    if(s[i]=='c') u=1,v=2;
    if(vis[bel[i<<1]]==1) putchar(u+'A'-1);
    else putchar(v+'A'-1);
  }
  return Back(true);
}

bool dfs(int x){
  if(x>=xp.size()){
    if(judge()) return true;
    return false;
  }
  b[xp[x]]=0; s[xp[x]]='a'; 
  if(dfs(x+1)) return true;
  s[xp[x]]='x'; b[xp[x]]=1; 
  if(dfs(x+1)) return true;
  return false;
}

int ddd;

int main(){
  scanf("%d%d",&n,&ddd);
  scanf("%s",s+1);
  for(int i=1;i<=n;i++)
    if(s[i]=='x') xp.push_back(i);
  scanf("%d",&m);
  for(int i=1;i<=m;i++){
    scanf("%d",&e[i].a);
    while((e[i].x=getchar())!='A' && e[i].x!='C' && e[i].x!='B');
    scanf("%d",&e[i].b);
    while((e[i].y=getchar())!='A' && e[i].y!='C' && e[i].y!='B');
    inn[e[i].a].push_back(e[i]);
    outt[e[i].b].push_back(e[i]);
  }
  if(!dfs(0)) puts("-1");
  return 0;
}

你可能感兴趣的:(2-sat,2-SAT)