牛客算法周周练8 A 小A买彩票

链接：https://ac.nowcoder.com/acm/contest/5803/A
来源：牛客网
 

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

小A最近开始沉迷买彩票，并且希望能够通过买彩票发家致富。已知购买一张彩票需要3元，而彩票中奖的金额分别为1,2,3,4元，并且比较独特的是这个彩票中奖的各种金额都是等可能的。现在小A连续购买了n张彩票，他希望你能够告诉他至少能够不亏本的概率是多少。

输入描述:

一行一个整数N，为小A购买的彩票数量一行一个整数N，为小A购买的彩票数量一行一个整数N，为小A购买的彩票数量

输出描述:

输出一个最简分数a/b，表示小A不亏本的概率。若概率为1，则输出1/1，概率为0，则输出0/1。输出一个最简分数a/b，表示小A不亏本的概率。 \\若概率为1，则输出1/1，概率为0，则输出0/1。输出一个最简分数a/b，表示小A不亏本的概率。若概率为1，则输出1/1，概率为0，则输出0/1。

示例1

输入

2

输出

3/8

备注:

0≤n≤30

思路：本以为是道数学题，结果找不到规律，最后发现是DP；

DP[i][j]表示第i个共j元时的情况数，那么转移方程就很明显了：dp[i][j] = dp[i-1][j-1] + dp[i-1][j-2] + dp[i-1][j-3] + dp[i-1][j-4]；

#include<iostream>
#include<algorithm>
#define ll long long 

using namespace std;

ll dp[1000][1000];

ll ksm(ll a, ll b)
{
	ll rul = 1;
	while(b)
	{
		if(b & 1) rul *= a;
		b >>= 1;
		a *= a;
	}
	return rul;
}

int main()
{
	dp[0][0] = 1;
	int n;
	cin>>n;
	for(int i=1; i<=n; i++)
	{
		for(int j=i; j<=4*i; j++)
		{
			for(int k=1; k<=4; k++)
			{
				if(j >= k) dp[i][j] += dp[i-1][j-k];
			}
		}
	}
	ll a = 0;
	for(int i=3*n; i<=4*n; i++)
		a += dp[n][i];
	ll b = ksm(4,n);
	ll gcd = __gcd(a,b);
	printf("%lld/%lld",a/gcd,b/gcd);
	return 0;
}