计蒜客 方程的解数（DFS）

问题描述

输出格式

输出一行，输出一个整数，表示方程的整数解的个数。

样例输入

3
100
1 2
-1 2
1 2

样例输出

104

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <string>
#include <math.h>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <sstream>
const int INF=0x3f3f3f3f;
typedef long long LL;
const int mod=1e9+7;
const double PI = acos(-1);
const double eps =1e-8;
#define Bug cout<<"---------------------"<<endl
const int maxn=1e5+10;
using namespace std;

int k[5],p[5]; 
int n,m,ans;
LL POW[150][5];//POW[i][j]表示i的j次方 

void init()//打表
{
    for(int i=1;i<=150;i++)
    {
        POW[i][0]=1;
        for(int j=1;j<=4;j++)
        {
            POW[i][j]=i*POW[i][j-1];
        }
    }
}

void DFS(int step,LL sum)
{
    if(step>n)
    {
        if(sum==0) ans++;
        return ;
    }
    for(int i=1;i<=m;i++)
        DFS(step+1,sum+k[step]*POW[i][p[step]]);
}

int main()
{
    #ifdef DEBUG
    freopen("sample.txt","r",stdin);
    #endif
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d %d",&k[i],&p[i]);
    init();//不要忘了 
    DFS(1,0);
    printf("%d
",ans);
    
    return 0;
}

-