hdu 6125 -- Free from square(状态压缩+分组背包)

题目链接

Problem Description

There is a set including all positive integers that are not more then n. HazelFan wants to choose some integers from this set, satisfying: 1. The number of integers chosen is at least 1 and at most k. 2. The product of integers chosen is 'free from square', which means it is divisible by no square number other than 1. Now please tell him how many ways are there to choose integers, module 10^9+7.

 

Input

The first line contains a positive integer T(1≤T≤5), denoting the number of test cases.
For each test case:
A single line contains two positive integers n,k(1≤n,k≤500).

 

Output

For each test case:
A single line contains a nonnegative integer, denoting the answer.

 

Sample Input

2

4 2

6 4

 

Sample Output

6

19

题意：从1~n中任意取1~K个数（同一个数不能用多次），这些数的乘积不能被任意一个数的平方整除（除了 1 ），求有多少种取法？

思路：可以从以上题意分析出，取的多个数不能有相同的质数因子。由于n<=500，所以一个数小于sqrt(n)的质数因子可能有多个，但大于sqrt(n)的质数因子只可能有一个。而小于sqrt(n)的质数有2 、3、5、7、11、13、17、19，一共8个，所以对这8个质数因子进行状压。然后就是背包DP，因为成绩不能含有 质数因子的平方，所以需要进行分组，将含有相同大于sqrt（n）的数放在一组，保证这样的数只能每次取一个，也就是分组背包。

代码如下：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int mod=1e9+;
const int N=;
int dp[N][];
int r[N],st[N];
int p[]={,,,,,,,};
vector<int>v[N];
 
int main()
{
    int T; cin>>T;
    while(T--)
    {
        int n,m; scanf("%d%d",&n,&m);
        for(int i=;i<N;i++)
        {
            v[i].clear();
            r[i]=i;
            st[i]=;
        }
        memset(dp,,sizeof(dp));
        dp[][]=;
        for(int i=;i<=n;i++)
        {
            for(int j=;j<;j++)
            {
                if(i%p[j]== && i%(p[j]*p[j])!=) st[i]|=<<j,r[i]/=p[j];
                else if(i%(p[j]*p[j])==){
                    st[i]=-; break;
                }
            }
        }
        for(int i=;i<=n;i++)
        {
            if(st[i]==-) continue;
            if(r[i]==) v[i].push_back(i);
            else v[r[i]].push_back(i);
        }
//        for(int i=1;i<=n;i++)
//        {
//            for(int j=0;j<v[i].size();j++)
//                cout<<v[i][j]<<" ";
//            cout<<endl;
//        }
        for(int i=;i<=n;i++)
        {
            if(st[i]==- || v[i].size()==) continue;
            for(int j=m-;j>=;j--)
            {
                for(int s=;s<;s++)
                {
                    for(int k=;k<v[i].size();k++)
                    {
                        int d=st[v[i][k]];
                        if((s&d)!=) continue;
                        dp[j+][s|d]=(dp[j+][s|d]+dp[j][s])%mod;
                    }
                }
            }
        }
        int ans=;
        for(int i=;i<=m;i++)
        {
            for(int j=;j<;j++)
                ans=(ans+dp[i][j])%mod;
        }
        printf("%d\n",ans);
    }
    return ;
}