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