HDU 6125 Free from square（数论+状压DP）

Description

从$1$到$n$中选取$k$个不同的数字使得乘积为无平方因子数，问方案数

Input

第一行一整数$T$表示用例组数，每组用例输入两个整数$n$个$k$$(1\le T\le 5,1\le n,k\le 500)$

Output

对于每组用例，输出方案数，结果模$10^9+7$

Sample Input

2
4 2
6 4

Sample Output

6
19

Solution

对于所选的$k$个数，大于$\sqrt{n}$的素因子每个数至多包含一个，把小于$\sqrt{n}$的$8$个素因子状压起来，每个数字就可以变成一个大于$\sqrt{n}$的因子加上一个状态表示其含有的小素因子，把这个数字归类到它的大素因子下，那么问题变为有若干堆物品，每个物品用一个$0$~$255$的状态表示，从每一堆中至多选一个物品且状态不交的方案数，一个简单$dp$即可解决，时间复杂度$O(256nk)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
#define maxn 505
#define mod 1000000007
int p[]={2,3,5,7,11,13,17,19};
int T,n,k,a[maxn],sta[maxn],dp[maxn][maxn];
vector<int>v[maxn];
void inc(int &a,int b)
{
    a=a+b>=mod?a+b-mod:a+b;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++)v[i].clear(),sta[i]=0;
        for(int i=1;i<=n;i++)
        {
            a[i]=i;
            for(int j=0;j<8;j++)
                if(a[i]%(p[j]*p[j])==0)sta[i]=-1;
                else if(a[i]%p[j]==0)a[i]/=p[j],sta[i]|=(1<<j);
        }
        for(int i=1;i<=n;i++)
            if(~sta[i])
            {
                if(a[i]==1)v[i].push_back(i);
                else v[a[i]].push_back(i);
            }
        memset(dp,0,sizeof(dp));
        dp[0][0]=1;
        for(int i=1;i<=n;i++)
            if(~sta[i]&&v[i].size())
                for(int j=k-1;j>=0;j--)
                    for(int s=0;s<256;s++)
                        for(int l=0;l<v[i].size();l++)
                        {
                            int t=sta[v[i][l]];
                            if(s&t)continue;
                            inc(dp[j+1][s|t],dp[j][s]);
                        }
        int ans=0;
        for(int i=1;i<=k;i++)
            for(int j=0;j<256;j++)
                inc(ans,dp[i][j]);
        printf("%d\n",ans);
    }    
    return 0;
}