NOIP模拟 乘积【状压dp+多组背包】

题目大意：

选择不超过k个n以内的正整数相乘，使乘积使一个无平方因子数，问有多少种取法？（每个数只能取一次）$1\le k,n\le 500$。

解题思路：

首先可以想到把有平方因子数的数删了。

那问题也就变成了使取得的数的质因数集合无交集。

如果质因数个数足够少，我们就可以状压记录每个质数取还是没取，但500的范围太大。

不过注意到每个数最多只会有一个比$\sqrt{n}$大的质因数，所以我们可以把所有数字按照含有哪一个比$\sqrt{n}$大的质因数分组，若没有则自己分一组，那么最后每组最多取一个数，问题也就变成了多组背包问题。

而$\sqrt{n}$以内的质因数只有8个，就可以状压了。

f[i][j][mask]表示计算完了前i组，共选择了j个数字，mask表示当前的质因数集合有哪些质数。但第一维就像01背包一样可以省去。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<set>
#define ll long long
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=505,mod=1e9+7;
const int p[8]={2,3,5,7,11,13,17,19};
int T,n,k;
int yz[N],g[N][N];
int f[N][1<<8];

void pre()
{
    for(int i=1;i<=500;i++)
    {
        for(int j=0;j<8;j++)
        {
            int tmp=i;
            if(tmp%p[j]==0)yz[i]|=1<<j,tmp/=p[j];
            if(tmp%p[j]==0)
            {
                yz[i]=-1;
                break;
            }
        }
    }
}

void Div()
{
    for(int i=1;i<=n;i++)
    {
        if(yz[i]==-1)continue;
        int tmp=i;
        for(int j=0;j<8;j++)
            while(tmp%p[j]==0)tmp/=p[j];
        if(tmp!=1)g[tmp][++g[tmp][0]]=i;
        else g[i][++g[i][0]]=i;
    }
}

void Add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}

int main()
{
    //freopen("lx.in","r",stdin);
    //freopen("lx.out","w",stdout);
    pre();
    T=getint();
    while(T--)
    {
        memset(g,0,sizeof(g));
        memset(f,0,sizeof(f)); 
        n=getint(),k=getint();
        Div();
        f[0][0]=1;
        for(int i=1;i<=n;i++)
            if(g[i][0])
            {
                for(int j=k-1;j>=0;j--)
                    for(int l=1;l<=g[i][0];l++)
                        for(int now=0,v=yz[g[i][l]];now<(1<<8);now++)
                            if(!(now&v))Add(f[j+1][now|v],f[j][now]);
            }
        ll ans=0;
        for(int i=1;i<=k;i++)
            for(int now=0;now<(1<<8);now++)
                ans=(ans+f[i][now])%mod;
        cout<<ans<<'\n';
    }
    return 0;
}