hdu 6327 - 期望DP+状态优化

题目链接：点击这里

 

解题思路:令id[i][j][k]表示最后三个数的gcd是i，最后两个数的gcd是j，最后一个数是k的状态值。这样表示的话状态就会被大大的压缩，而且我们的压缩方法也是正确的。因为题目要求的是gcd而不是他的值，所以不管最后三个数是什么，如果他们的i，j，k相同。说明他们的价值就是一样的。这样就可以把一个100*100*100的状态数压缩成大约1500的状态数。那么当p位置转移到p+1时，id值的改变状态也可以预处理出来。

设f[p][i][j][k]表示到达p位置最后三个数的gcd是i，最后两个数的gcd是j，最后一个数是k的数学期望，那么

  f[p+1][i][j][k] =  f[p][i][j][k] * 最后四个数的相应gcd的v值

 

#include<bits/stdc++.h>
using namespace std;
const int mx = 1e2 + 5;
const int mod = 1e9 + 7;
typedef long long ll;
int n,a[mx],N,gcd[mx][mx],b[mx];
int id[mx][mx][mx],w[1500][mx];
int v[1500][mx];
ll f[mx][1500];
ll qpow(ll x,ll y)
{
    ll ans = 1;
    while(y){
        if(y&1) ans = ans*x%mod;
        y >>= 1;
        x = x*x%mod;
    } 
    return ans;
}
int main()
{
    //for(int i=0;i<=N;i++)
    //for(int j=1;j<=N;j++) sum[i] += num[i][j];
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&N);
        memset(f,0,sizeof(f));
        for(int i=1;i<=n;i++) scanf("%d",a+i);
        for(int i=1;i<=N;i++) scanf("%d",b+i);
        for(int i=1;i<=N;i++)
        for(int j=1;j<=N;j++) gcd[i][j] = __gcd(i,j);
        int cnt = 0;
        for(int i=1;i<=N;i++) 
            for(int j=i;j<=N;j+=i)
                for(int k=j;k<=N;k+=j) id[i][j][k] = ++cnt;
        for(int i=1;i<=N;i++)
            for(int j=i;j<=N;j+=i)
                for(int k=j;k<=N;k+=j){
                    int x = id[i][j][k];
                    for(int y=1;y<=N;y++)
                    w[x][y] = id[gcd[j][y]][gcd[k][y]][y],v[x][y] = b[gcd[i][y]];
                }
        for(int i=1;i<=N;i++){
            for(int j=1;j<=N;j++){
                for(int k=1;k<=N;k++){
                    if(a[1]&&i!=a[1]) continue;
                    if(a[2]&&j!=a[2]) continue;
                    if(a[3]&&k!=a[3]) continue;
                    int u = gcd[gcd[i][j]][k];
                    f[3][id[u][gcd[j][k]][k]]++;
                }
            }
        }
        for(int i=3;i<n;i++){
            for(int j=1;j<=cnt;j++){
                if(f[i][j])
                for(int k=1;k<=N;k++){
                    if(a[i+1]&&k!=a[i+1]) continue;
                    f[i+1][w[j][k]] = (f[i+1][w[j][k]]+f[i][j]*v[j][k])%mod;
                }
            }
        } 
        ll ans = 0,c = 0;
        for(int i=1;i<=cnt;i++) ans = (ans+f[n][i])%mod;
        for(int i=1;i<=n;i++) if(!a[i]) c++;
        printf("%lld\n",ans*qpow(qpow(N,c),mod-2)%mod);
    }
    return 0;
}