Codeforces Gym 100339I Sum vs Product 가지치기 검색

제목 대의:
n<=500을 드리는 거예요.
그리고 중복원소의 집합을 구성하여 그 중 모두 n개의 원소가 있고 모든 원소와 적체가 같은 집합이 몇 개냐고 물었다
사용하는 요소는 양의 정수여야 합니다.
대략적인 사고방식:
사실은 가지치기 검색이에요.
이러한 집합 수량이 틀림없이 많지 않다는 것을 감안하여 가지를 잘라서 검색하면 된다
평소에 검색어를 많이 안 쓰는데...그래도 문제풀이를 써봤는데...
코드는 다음과 같습니다.
Result  :  Accepted     Memory  :  184 KB     Time  :  60 ms
/*
 * Author: Gatevin
 * Created Time:  2015/9/2 15:04:39
 * File Name: fz.cpp
 */
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;

int n;
int ans;

void dfs(int now, int sum , int pro, int num)
{
    if(sum < pro && num > 1) return;
    if(now == 0)
    {
        ans += (sum == pro);
        return;
    }
    if(num == 1)
    {
        dfs(now - 1, sum + 1, pro, 1);
        for(int k = num + 1; sum + k >= pro*k && k <= 500; k++)
            dfs(now - 1, sum + k, pro*k, k);
    }
    else
    {
        for(int k = num; sum + k >= pro*k && k <= 500; k++)
            dfs(now - 1, sum + k, pro*k, k);
    }
}

int main()
{
    freopen("sump.in", "r", stdin);
    freopen("sump.out", "w", stdout);
    ans = 0;
    scanf("%d", &n);
    dfs(n, 0, 1, 1);
    printf("%d
", ans); return 0; }

좋은 웹페이지 즐겨찾기