LibreOJ2305. 「NOI2017」游戏【2-SAT】

2305. 「NOI2017」游戏

先不考虑x的地图。

这样一看，好像需要3-SAT，其实不需要，我们将每幅图拆成两个点就可以了。

但是x就有点恶心了，还好个数不多，所以直接DFS就可以了。

数组开小了导致T飞QAQ

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=5e4+5,MAXE=1e5+5;
int n,d,m;char ch[MAXN];bool f;
struct Edge{
	int tot,lnk[MAXN*2],nxt[MAXE<<1],son[MAXE<<1];
	void Clear(){memset(lnk,0,sizeof(lnk));tot=0;}
	void Add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
}E;
#include<cctype>
int read(){
	int ret=0;char ch=getchar();bool f=1;
	for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');
	for(; isdigit(ch);ch=getchar()) ret=ret*10+ch-48;
	return f?ret:-ret;
}
bool Ask(int x,char c){
	if(ch[x]=='A'||ch[x]=='B') return c=='C';
	if(ch[x]=='C') return c=='B';
}
char ret(int x,bool ff){
	if(ch[x]=='A') return ff?'C':'B';
	if(ch[x]=='B') return ff?'C':'A';
	if(ch[x]=='C') return ff?'B':'A';
}
struct xcw{int x1,x2;char c1,c2;}a[MAXE];
void Build(){
	E.Clear();
	for(int i=1;i<=m;i++){
		int x1=a[i].x1,x2=a[i].x2;char c1=a[i].c1,c2=a[i].c2;
		if(ch[x1]==c1) continue;
		if(ch[x2]==c2){E.Add(x1-1<<1|Ask(x1,c1),x1-1<<1|(!Ask(x1,c1)));continue;}
		E.Add(x1-1<<1|Ask(x1,c1),x2-1<<1|Ask(x2,c2));
		E.Add(x2-1<<1|(!Ask(x2,c2)),x1-1<<1|(!Ask(x1,c1)));
	}
}
int cnt,Dfn[MAXN*2],Low[MAXN*2],Stk[MAXN*2],Top,Fa[MAXN*2],tim;bool InStk[MAXN*2];
void Tarjan(int x){
	Dfn[x]=Low[x]=++tim;Stk[++Top]=x;InStk[x]=1;
	for(int j=E.lnk[x];j;j=E.nxt[j])
	if(!Dfn[E.son[j]]) Tarjan(E.son[j]),Low[x]=min(Low[x],Low[E.son[j]]);
	else if(InStk[E.son[j]]) Low[x]=min(Low[x],Dfn[E.son[j]]);
	if(Low[x]==Dfn[x]){++cnt;do{InStk[Stk[Top]]=0;Fa[Stk[Top]]=cnt;}while(Stk[Top--]!=x);}
}
void INIT(){for(int i=0,END=2*n;i<END;i++) Dfn[i]=Low[i]=Fa[i]=0;tim=0,cnt=0;}
void Work(){
	INIT();Build();
	for(int i=0,END=2*n;i<END;i++) if(!Dfn[i]) Tarjan(i);
	for(int i=1;i<=n;i++) if(Fa[i-1<<1]==Fa[i-1<<1|1]&&Fa[i-1<<1]!=0) return;
	f=1;
	for(int i=1;i<=n;i++)
	if(Fa[i-1<<1]<Fa[i-1<<1|1]) printf("%c",ret(i,0));else printf("%c",ret(i,1));
}
void DFS(int x){
	if(f) return;
	if(x>n){Work();return;}
	if(ch[x]!='X'){DFS(x+1);return;}
	ch[x]='A';DFS(x+1);ch[x]='B';DFS(x+1);
}
int main(){
	n=read(),d=read();scanf("%s",ch+1);
	for(int i=1;i<=n;i++) ch[i]=ch[i]-'a'+'A';
	m=read();
	for(int i=1;i<=m;i++){
		int x1,x2;char c1,c2;
		scanf("%d%*c%c%d%*c%c",&x1,&c1,&x2,&c2);
		a[i]=(xcw){x1,x2,c1,c2};
	}
	DFS(1);
	if(!f) printf("-1\n");
	return 0;
}