[NOI2017]游戏

题意

有三种数,每个位置只能填一种数,有$d\in[0,8]$个位置有三种选择,其他位置只有两种选择

给你一些限制,表示第$i$个位置选了某种数,那么第$j$个位置就只能选规定的数

输出一组合法的选数方案,无解输出$-1$

---

题解

①考虑$d=0$,那么就是每个位置都只有两种选择和一堆限制

那么这就是一道$2-sat$的裸题了

假设限制的四元组是$(i,h_i,j,h_j)$,令$u=[$第$i$个位置填$h_i$的编号$]$,$v=[$第$j$个位置填$h_j$的编号$]$,那么连边$(u,v),(v',u')$

注意如果$v$是不存在的(也就是第$j$个位置不能填$h_j$)那么$u$就不能选,那么就连边$(u,u')$

然后如果$u$是存在的话,这个限制就废了(可以稍微快一点$?$)

输出方案就是建反图然后拓扑排序然后染色

其实可以直接$for$一遍就好了,设$belong_i$表示$i$所属的强连通分量

按照原来的做法,如果反图$belong_i$的拓扑序小于$belong_{i'}$那么这个位置就选能选的第一种数,否则就是第二种

考虑到$tarjan$求出来的强连通分量就是按照反图的逆序给出的

原来的方法等价一下就是如果$belong_i<belong_{i'}$就选这个位置能选的第一种,否则就是第二种

②考虑$d\gt0$怎么做,就是有些位置是$3-sat?$

显然不可能是$3-sat$,毕竟$NPC$

一个简单的想法就是$3^d$的$dfs$枚举这个位置填什么

而且这样也没法优化了,考虑正难则反,枚举这一位不填什么

比如:

$1.$假设这个位置不能选$A$数,那么就只能选$B,C$数;

$2.$假设这个位置不能选$B$数,那么就只能选$A,C$数;

$3.$假设这个位置不能选$C$数,那么就只能选$A,B$数;

我们可以发现第$3$种情况已经在前两种里面讨论过了,这样我们就能做到$2^d$了

总时间复杂度$O(2^d(n+m))$

#include<bits/stdc++.h>
#define fp(i,a,b) for(register int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(register int i=a,I=b-1;i>I;--i)
#define go(u) for(register int i=fi[u],v=e[i].to;i;v=e[i=e[i].nx].to)
#define file(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
using namespace std;
char ss[1<<17],*A=ss,*B=ss;
inline char gc(){return A==B&&(B=(A=ss)+fread(ss,1,1<<17,stdin),A==B)?-1:*A++;}
template<class T>inline void sd(T&x){
    char c;T y=1;while(c=gc(),(c<48||57<c)&&c!=-1)if(c==45)y=-1;x=c-48;
    while(c=gc(),47<c&&c<58)x=x*10+c-48;x*=y;
}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
template<class T>inline void we(T x){
    if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
const int N=1e5+5,M=2e5;
typedef int arr[N];
struct eg{int nx,to;}e[M];
struct da{int x,y;}a[N],b[N];
int n,m,ce,dft,Cnt,Top,pos[N][3];arr p,f,nt,fi,bl,dfn,low,vis,S,Ch;
void tarjan(int u){
    dfn[u]=low[u]=++dft;vis[u]=1,S[++Top]=u;
    go(u)if(!dfn[v])tarjan(v),cmin(low[u],low[v]);
        else if(vis[v])cmin(low[u],dfn[v]);
    if(dfn[u]==low[u]){
        ++Cnt;int v;
        do{vis[v=S[Top--]]=0,bl[v]=Cnt;}while(Top&&v^u);
    }
}
inline void add(int u,int v){e[++ce]={fi[u],v},fi[u]=ce;}
#define clr(x) memset(x,0,sizeof x)
inline void sol(){
    ce=Cnt=dft=0;clr(fi),clr(dfn),clr(low);
    fp(i,1,n)nt[nt[i]=i+n]=i;
    fp(i,1,m)if(p[a[i].x]^a[i].y){
        int u=pos[a[i].x][a[i].y],v=pos[b[i].x][b[i].y];
        if(p[b[i].x]==b[i].y)add(u,nt[u]);
        else add(u,v),add(nt[v],nt[u]);
    }
    fp(i,1,2*n)if(!dfn[i])tarjan(i);
    fp(i,1,n)if(bl[i]==bl[i+n])return;
        else nt[bl[i+n]]=bl[i],nt[bl[i]]=bl[i+n];
    fp(i,1,n)putchar('A'+Ch[i+(bl[i]<bl[i+n]?0:n)]);
    exit(0);
}
void dfs(int d){
    if(d==f[0]+1)return sol();int i=f[d];
    p[i]=0,pos[i][Ch[i]=1]=i;dfs(d+1);pos[i][1]=0;
    p[i]=1,pos[i][Ch[i]=0]=i;dfs(d+1);
}char s[N];
int main(){
    #ifndef ONLINE_JUDGE
        file("s");
    #endif
    scanf("%d%*d%s",&n,s+1);
    fp(i,1,n){
        p[i]=s[i]-'a';
        if(p[i]>2)f[++f[0]]=i,pos[i][2]=i+n,Ch[i+n]=2;
        else{
            if(p[i]==0)pos[i][1]=i,pos[i][2]=i+n,Ch[i]=1,Ch[i+n]=2;
            else if(p[i]==1)pos[i][0]=i,pos[i][2]=i+n,Ch[i]=0,Ch[i+n]=2;
            else pos[i][0]=i,pos[i][1]=i+n,Ch[i]=0,Ch[i+n]=1;
        }
    }scanf("%d",&m);int k=0;char c1,c2;
    fp(i,1,m){
        scanf("%d %c%d %c",&a[i].x,&c1,&b[i].x,&c2);
        if(s[a[i].x]==c1)continue;
        a[++k]={a[i].x,c1-'A'},b[k]={b[k].x,c2-'A'};
    }m=k;dfs(1);puts("-1");
return 0;
}

如果你不想写$spj$的话戳这里$(by\ $$loj$ ) ) <script type="math/tex" id="MathJax-Element-64">)</script>