SPOJ AMR10I Dividing Stones --DFS

题意：给n个石头，分成一些部分（最多n部分，随便分），问分完后每部分的数量的乘积有多少种情况。

分析：可以看出，其实每个乘积都可以分解为素数的乘积，比如乘积为4，虽然可以分解为4*1，但是更可以分解为2*2*1，所以就可以枚举素因子来分解，dfs即可。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <set>
#define ll long long
using namespace std;
#define N 100007

int prime[22] = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71};
int n,p;
set<ll> ans;

void dfs(int ind,int now,ll num)
{
    ans.insert(num);
    if(now < prime[ind])
        return;
    dfs(ind,now-prime[ind],num*prime[ind]%p);  //分解这个
    dfs(ind+1,now,num);  //不分解这个
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&p);
        ans.clear();
        dfs(0,n,1);
        printf("%d\n",ans.size());
    }
    return 0;
}

View Code