川大oj 4426

最新推荐文章于 2021-12-13 01:32:37 发布

tjqACM

最新推荐文章于 2021-12-13 01:32:37 发布

阅读量839

点赞数

文章标签： dp

本文链接：https://blog.csdn.net/tjqACM/article/details/44964237

版权

题目大意：只能使用3的x次方的和，并且每个次方的数最多用三次，求组合成和值为n的方法数

比赛的时候没有想到用dp的方法来解决，因为n很大，不好转移，就一直在哪里对公式，导致最后GG，下来自己仔细想了一下，这个数必须由3的x次组成，因此只需将n转换为三进制的数，然后从高为枚举，每位的状态只能为0,1,2,3,4。可以为4是因为他可以往后面转移，因此状态转移可一很容易的写出来，详见代码：

#include<cstdio>
#include<stack>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;

stack<int>s;
int a[100],dp[100][10];
int k;
void slove(LL num)
{
    while(num)
    {
        LL x=num%3;
        num=num/3;
        s.push(x);
    }
    k=0;
    while(!s.empty())
    {
        a[k++]=s.top();
        s.pop();
    }
}
int main()
{
    int T;
    LL n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld",&n);
        slove(n);
        memset(dp,0,sizeof(dp));
        dp[0][a[0]]=1;
        for(int i=1;i<k;i++)
        {
            if(a[i]==0)
            {
                dp[i][0]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2]+dp[i-1][3];
                dp[i][3]=dp[i-1][1]+dp[i-1][2]+dp[i-1][3]+dp[i-1][4];
            }
            else if(a[i]==1)
            {
                dp[i][1]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2]+dp[i-1][3];
                dp[i][4]=dp[i-1][1]+dp[i-1][2]+dp[i-1][3]+dp[i-1][4];
            }
            else if(a[i]==2)
            {
                dp[i][2]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2]+dp[i-1][3];
            }
        }
        int ans=0;
        for(int i=0;i<=3;i++)
        {
            ans+=dp[k-1][i];
        }
        printf("%d\n",ans);
    }
    return 0;
}