HDU5456 Matches Puzzle Game (记忆化搜索)

题目连接:http://acm.hdu.edu.cn/showproblem.php?pid=5456


题目大意:给定n(n<=500)根火柴棒,求问用n根火柴能组成的a - b = c 的格式所有可能情况数。


解题思路:把a-b=c的式子转化成b+c = a,然后对其进行记忆化搜索。


dp[i][carry][B][C]表示剩余i根火柴棒时,(carry)是否有进位,是否在b放置,是否在c放置。


/* ***********************************************
┆  ┏┓   ┏┓ ┆
┆┏┛┻━━━┛┻┓ ┆
┆┃       ┃ ┆
┆┃   ━   ┃ ┆
┆┃ ┳┛ ┗┳ ┃ ┆
┆┃       ┃ ┆
┆┃   ┻   ┃ ┆
┆┗━┓ 马 ┏━┛ ┆
┆  ┃ 勒 ┃  ┆      
┆  ┃ 戈 ┗━━━┓ ┆
┆  ┃ 壁     ┣┓┆
┆  ┃ 的草泥马  ┏┛┆
┆  ┗┓┓┏━┳┓┏┛ ┆
┆   ┃┫┫ ┃┫┫ ┆
┆   ┗┻┛ ┗┻┛ ┆
************************************************ */
//#pragma comment(linker, "/STACK:102400000,102400000")
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <bitset>
using namespace std;

#define rep(i,a,b) for (int i=(a),_ed=(b);i<=_ed;i++)
#define per(i,a,b) for (int i=(b),_ed=(a);i>=_ed;i--)
#define pb push_back
#define mp make_pair
const int inf_int = 2e9;
const long long inf_ll = 2e18;
#define inf_add 0x3f3f3f3f
#define mod 1000000007
#define LL long long
#define ULL unsigned long long
#define MS0(X) memset((X), 0, sizeof((X)))
#define SelfType int
SelfType Gcd(SelfType p,SelfType q){return q==0?p:Gcd(q,p%q);}
SelfType Pow(SelfType p,SelfType q){SelfType ans=1;while(q){if(q&1)ans=ans*p;p=p*p;q>>=1;}return ans;}
#define Sd(X) int (X); scanf("%d", &X)
#define Sdd(X, Y) int X, Y; scanf("%d%d", &X, &Y)
#define Sddd(X, Y, Z) int X, Y, Z; scanf("%d%d%d", &X, &Y, &Z)
#define reunique(v) v.resize(std::unique(v.begin(), v.end()) - v.begin())
#define all(a) a.begin(), a.end()
#define   mem(x,v)      memset(x,v,sizeof(x))
typedef pair<int, int> pii;
typedef pair<long long, long long> pll;
typedef vector<int> vi;
typedef vector<long long> vll;
inline int read(){int ra,fh;char rx;rx=getchar(),ra=0,fh=1;while((rx<'0'||rx>'9')&&rx!='-')rx=getchar();if(rx=='-')fh=-1,rx=getchar();while(rx>='0'&&rx<='9')ra*=10,ra+=rx-48,rx=getchar();return ra*fh;}

int n,m;
LL dp[505][2][2][2];
int bit[10] = {6, 2, 5, 5, 4, 5, 6, 3, 7, 6};

void add(LL &x,LL y)
{
    x = (x + y) % m;
}

LL dfs(int n,bool carry,int B,int C)
{
    LL &res = dp[n][carry][B][C];
    if(~res) return res;
    if(n==0)
    {
        if(!carry && !B && !C) return res = 1;
        return res = 0;
    }
    res = 0;
    if(B && C)
    {
        for(int i=0;i<10;i++)
        {
            for(int j=0;j<10;j++)
            {
                int sum = bit[i] + bit[j] + bit[(i+j+carry)%10];
                if(sum>n)continue;
                bool nx = i+j+carry >= 10;
                add(res,dfs(n-sum,nx,1,1));
                if(i) add(res,dfs(n-sum,nx,0,1));
                if(j) add(res,dfs(n-sum,nx,1,0));
                if(i&&j) add(res,dfs(n-sum,nx,0,0));
            }
        }
    }
    else if(B)
    {
        for(int i=0;i<10;i++)
        {
            int sum = bit[i] + bit[(i+carry)%10];
            if(sum>n)continue;
            bool nx = i+carry >=10;
            add(res,dfs(n-sum,nx,1,0));
            if(i) add(res,dfs(n-sum,nx,0,0));
        }
    }
    else if(C)
    {
        for(int i=0;i<10;i++)
        {
            int sum = bit[i] + bit[(i+carry)%10];
            if(sum>n)continue;
            bool nx = i+carry >=10;
            add(res,dfs(n-sum,nx,0,1));
            if(i) add(res,dfs(n-sum,nx,0,0));
        }
    }
    else if(carry && n==bit[1]) res = 1;
    return res;
}



int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	//ios::sync_with_stdio(0);
	//cin.tie(0);
	int t = read();
	int cas = 1;
	while(t--)
    {
        n = read(), m = read();
        n -= 3;
        memset(dp,-1,sizeof dp);
        printf("Case #%d: %I64d\n", cas++, dfs(n, 0, 1, 1));

    }


	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值