交通管控

D06發表於2024-08-06
  • 用四進製表示三進位制數時,需要預處理出所有合法的狀態,否則會因多出許多冗餘狀態而導致TLE
  • 在原始碼的基礎上修改時很容易出錯:1~tot
點選檢視程式碼
#include <bits/stdc++.h>
using namespace std;
int f[2][(1<<20)+5];
bool g[2][(1<<20)+5];
int h[(1<<20)+5];
int n,k,mod;
int get(int x,int id)
{
	return ((x>>(2*id))&3);
}
int change(int x,int id,int va)
{
	int v=get(x,id),tmp=v;
	v+=va;
	v%=3;
	return x-tmp*(1<<(2*id))+v*(1<<(2*id));
}
int calc(int x,string s)
{
	for(int i=0;i<k;i++)
	{
		if(s[i]=='+')
		{
			x=change(x,i,2);
		}
		else if(s[i]=='-')
		{
			x=change(x,i,1);
		}
	}
	return x;
}
const char c[]={'A','B','C'};
void shuchu(int x)
{
	for(int i=0;i<k;i++)
	{
		cout<<c[get(x,i)];
	}
	cout<<" "<<f[n&1][x]<<endl;
}
int st,tot;
void dfs1(int n1)
{
	if(n1==k)
	{
		h[++tot]=st;
	}
	else
	{
		for(int i=0;i<3;i++)
		{
			st+=((1<<(2*n1))*i);
			dfs1(n1+1);
			st-=((1<<(2*n1))*i);
		}
	}
}
void dfs2(int n1)
{
	if(n1==k)
	{
		if(g[n&1][st]==true)
		{
			shuchu(st);
		}
		f[n&1][st]=f[n&1^1][st]=0;
		g[n&1][st]=g[n&1^1][st]=false;
	}
	else
	{
		for(int i=0;i<3;i++)
		{
			st+=((1<<(2*n1))*i);
			dfs2(n1+1);
			st-=((1<<(2*n1))*i);
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n>>k>>mod;
		f[0][0]=1;
		g[0][0]=true;
		tot=0;
		dfs1(0);
		for(int i=1;i<=n;i++)
		{
			string s;
			cin>>s;
			for(int j=1;j<=tot;j++)
			{
				f[i&1][h[j]]=(f[i&1^1][h[j]]+f[i&1^1][calc(h[j],s)])%mod;
				g[i&1][h[j]]=(g[i&1^1][h[j]]|g[i&1^1][calc(h[j],s)]);
			}
		}
		dfs2(0);
	}
	return 0;
}

相關文章