题目概要:
小强是一个农场主,农场里有n头牛,每头牛有着独一无二的体重,每一头牛的颜色可能跟是m种颜色其中的一种,小强带了一些牛(可能为0个)出来吃草。你需要回答出小强带出来的牛的组合一共有多少种可能?
输入示例:
3 2
输出示例
27
分析示例:
牛的数量 组合数
0 1
1 6
2 12
3 8
我自己写的代码:
/*====================================================
* Copyright (C) 2020 All rights reserved.
*
* 文件名称:test.c
* 创 建 者:hlj 520@qq.com
* 创建日期:2020年07月31日
* 描 述:
================================================================*/
#include <stdio.h>
int di_gui(int m)
{
if(m == 0)
return 0;
if(m == 1)
return 1;
else
return m*di_gui(m-1);
}
int test(int n,int m)
{
int i = 0,j = 0;
int temp = 1;
int temp1 = 0;
int temp2 = 1;
int temp3 = 0;
int ret = n;
for(i = 1;i <= n;i++)
{
for(j = 0;j < i;j++)
{
temp = temp * (ret-j);
}
temp1 = di_gui(i);
temp2 = temp2*m;
printf("temp = :%d\n",temp);
printf("temp1 = :%d\n",temp1);
printf("temp2 = :%d\n",temp2);
temp3 += temp2*(temp/temp1);
temp = 1;
}
return temp3;
}
int main(int argc, char *argv[])
{
int n,m;
scanf("%d %d",&n,&m);
int ret;
ret = test(n,m);
printf("%d\n",ret+1);
return 0;
}
输出示例一:
输出示例2:
以下是在牛客上看到的一个大佬写的c++版,真就高下立断。。。
/*====================================================
* Copyright (C) 2020 All rights reserved.
*
* 文件名称:test1.cpp
* 创 建 者:hlj 520@qq.com
* 创建日期:2020年08月01日
* 描 述:
================================================================*/
#include <bits/stdc++.h>
using namespace std;
const int debug = 1;
const int size = 5000 + 10;
const int INF = INT_MAX>>1;
typedef long long ll;
ll quickpow_mod(ll a, ll b, ll m) {
a %= m;
ll res = 1;
while (b > 0) {
if (b & 1) res = res * a % m;
a = a * a % m;
b >>= 1;
}
return res;
}
int main(){
// std::ios::sync_with_stdio(false);cin.tie(0);
ll n, m;
cin >> n >> m;
ll mod = 1e9+7;
cout << quickpow_mod(m+1, n, mod) << endl;
return 0;
}
输出示例1:
输出示例2:
牛客上还有其他优化非常好的代码,有兴趣可以自己去查一查。