洛谷 P8557:炼金术(Alchemy)← 快速幂 + 组合

【题目来源】

https://www.luogu.com.cn/problem/P8557

【题目描述】

铃是一个爱玩游戏的女孩子。

她在游戏中想要炼制一种稀有合金 ------ 这需要 n 种金属来合成。

她准备好矿石后建造了 k 个不同的熔炉,当熔炉启动时,会随机炼出这 n 种金属中的一些(也可能什么都没有)。

如果把每个熔炉炼出的金属收集起来,有了全部 n 种金属,就能造出合金了。澪对此很好奇,对铃说:「我考考你,有多少种情况可以炼出合金呢?」这个简单的问题铃很快就会做了,你能求出结果吗?

答案可能很大,请对 998244353 取模(即除以 998244353 的余数)后输出。

【输入格式】

输入一行两个正整数 n,k。

【输出格式】

输出一行一个整数,表示答案。​​​​​​​

【输入样例】
233 123

【输出样例】
81633405

【数据范围】
对于 30% 的数据,1≤n,k≤10;
对于 80% 的数据,1≤n,k≤10^6;
对于 100% 的数据,1≤n,k≤10^9。

【算法分析】
● 对于任意一种金属,它可以被任意 1∼k 个熔炉所造出来,方案总数为:

C(k,1)+C(k,2)+C(k,3)+⋯⋯+C(k,k−1)+C(k,k)=2^k-1

现在有 n 种金属,每一种金属有 2^k-1 种方法,总方案数为 (2^k-1)^n,配上快速幂模板即可。

● 快速幂经典代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
 
typedef long long LL;
 
int fastPow(LL a,LL n) {
    LL ans=1;
    while(n){
        if(n & 1) ans=ans*a;
        n>>=1;
        a=a*a;
    }
    return ans;
}
 
int main() {
    int a,n;
    cin>>a>>n;
    cout<<fastPow(a,n)<<endl;
}
 
/*
in:6 8
out:1679616
*/

● 带取模的快速幂代码
计算过程中,为了防止溢出,需要进行"取模"运算,其运算规则如下:

(a+b)%p=(a%p+b%p)%p
(a-b)%p=(a%p-b%p)%p
(a*b)%p=(a%p*b%p)%p

【算法代码】

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int p=998244353;

LL fastPow(LL a,LL n) {
    LL ans=1;
    while(n) {
        if(n & 1) ans=ans*a%p;
        a=a*a%p;
        n>>=1;
    }
    return ans;
}

int main() {
    int n,k;
    scanf("%d%d",&n,&k);
    printf("%lld",fastPow(fastPow(2,k)-1,n));

    return 0;
}

/*
in:233 123
out:81633405
*/

【参考文献】

https://blog.csdn.net/hnjzsyjyj/article/details/143135845

相关推荐
汉克老师1 个月前
GESP2026年6月认证C++八级( 第三部分编程题(2、堆石子))精讲
c++·排列组合·快速幂·费马小定理·gesp8级·逆元·组合数计算
Q741_1474 个月前
每日一题 力扣 3655. 区间乘法查询后的异或 II 模拟 分治 乘法差分法 快速幂 C++ 题解
c++·算法·leetcode·模拟·快速幂·分治·差分法
Fcy6484 个月前
算法基础详解(六)倍增思想与离散化思想
算法·快速幂·离散化·倍增算法
芜湖xin7 个月前
【题解-Acwing】AcWing 5579. 增加模数(TLE)
算法·快速幂
让我们一起加油好吗1 年前
【基础算法】倍增
数学·算法·快速幂·洛谷·倍增
hnjzsyjyj2 年前
洛谷 P1226:【模板】快速幂
快速幂
hnjzsyjyj2 年前
AcWing 3625:幂次方 ← 快速幂
快速幂
Kent_J_Truman2 年前
【魔法 / NOI】
算法·动态规划·快速幂·矩阵乘法
Czi橙2 年前
玩玩快速冥(LeetCode50题与70题以及联系斐波那契)
java·算法·快速幂·斐波那契