HDU6327 Problem I. Random Sequence

补题的时候一开始想了个假算法，我预处理出有4个数字都是空的对答案的贡献anum4,bnum4，3个数字为空，一个数字为i,anum3[i],bnum4[i]，然后很膨胀地写到一半发现数字是要连续的，上一个空位确定后会对当前空位确定有可能产生影响。然后就不知道怎么办了，看了claris题解，存储的是f[i][x][y][z]表示到了第i位，z=a[i-1],y=gcd(a[i-1],a[i-2])，x=gcd(y,a[i-3])的情况对A有多少贡献。看似简单然而很不好写，于是对着claris究极压行的代码思考人生，claris机制得给所有x,y,z情况标了号，经过他的计算只有1500中情况，然后用g[s][y]表示x,y,z的情况编号后面一位是y时，转移到的编号，w[s][y]表示要乘的数字。然后进行转移计算出A，最后统计一遍B，有0的位置就乘以m。

#include<cstdio>
#include<cstring>
const int mod=1e9+7;
#define maxl 110
#define maxs 1500

int n,m,cnt,ans,numa,numb;
int a[maxl],v[maxl];
int gcd[maxs][maxl],g[maxs][maxl],w[maxs][maxl];
int f[maxl][maxs];
int id[maxl][maxl][maxl];

inline int cgcd(int a,int b)
{
	if(b==0)
		return a;
	else
		return cgcd(b,a%b);
}

inline int qp(int a,int b)
{
	int ans=1,cnt=a;
	while(b)
	{
		if(b&1)
			ans=(1ll*ans*cnt)%mod;
		cnt=(1ll*cnt*cnt)%mod;
		b>>=1;
	}
	return ans;
}

inline void prework()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<=m;i++)
		scanf("%d",&v[i]);
	for(int i=1;i<=m;i++)
		for(int j=1;j<=m;j++)
			gcd[i][j]=cgcd(i,j);
	cnt=0;
	for(int i=1;i<=m;i++)
		for(int j=i;j<=m;j+=i)
			for(int k=j;k<=m;k+=j)
				id[i][j][k]=++cnt;
	int x,y;
	for(int i=1;i<=m;i++)
		for(int j=i;j<=m;j+=i)
			for(int k=j;k<=m;k+=j)
			{
				x=id[i][j][k];
				for(y=1;y<=m;y++)
				{
					g[x][y]=id[gcd[j][y]][gcd[k][y]][y];
		        	w[x][y]=v[gcd[i][y]];
				}
			}
}

inline void mainwork()
{
	for(int i=1;i<=n;i++)
		for(int j=1;j<=cnt;j++)
			f[i][j]=0;
	int x,y,z,s;
	for(int i=1;i<=m;i++)
	{
		if(a[1] && a[1]!=i) continue;
		for(int j=1;j<=m;j++)
		{
			if(a[2] && a[2]!=j) continue;
			for(int k=1;k<=m;k++)
			{
				if(a[3] && a[3]!=k) continue;
				x=gcd[gcd[i][j]][k];y=gcd[j][k];z=k;
				s=id[x][y][z];
				f[3][s]=(f[3][s]+1)%mod;
			}
		}
	}
	for(int i=3;i<n;i++)
		for(int j=1;j<=cnt;j++)
		if(f[i][j])
			for(int k=1;k<=m;k++)
			{
      			if(a[i+1]&&k!=a[i+1])continue;
      			f[i+1][g[j][k]]=(f[i+1][g[j][k]]+1ll*f[i][j]*w[j][k]%mod)%mod;
    		}
    numa=0;numb=1;
    for(int j=1;j<=cnt;j++)
		numa=(numa+f[n][j])%mod;
    for(int i=1;i<=n;i++)
	if(!a[i])
		numb=1ll*numb*m%mod;
	ans=(1ll*numa*qp(numb,mod-2))%mod;
}

inline void print()
{
	printf("%d\n",ans);
}

int main()
{
	int t;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}