计算系数 Editional

计算系数

https://ac.nowcoder.com/acm/problem/16596

原题链接:https://ac.nowcoder.com/acm/problem/16596

题目描述:
给定一个多项式(ax+by)k,请求出多项式展开后xnym项的系数。
输入描述:
共一行,包含5个整数,分别为a,b,k,n,m,每两个整数之间用一个空格隔开。
输出描述:
输出共1行,包含一个整数,表示所求的系数,这个系数可能很大,输出对10007取模后的结果。
示例1
输入
1 1 3 1 2
输出
3
备注:
对于30%的数据,有0≤k≤10;
对于50%的数据,有a=1,b=1;
对于100%的数据,有0≤k≤1,000,0≤n,m≤k,且n+m=k,0≤a,b≤1,000,000。

记忆化的必要性
易得本题公式:ans = C(k,n) a^n b^m
若我们维护杨辉三角,就有O(k^2)的时空复杂度。
这样太慢了。
我们进一步观察可得:
ans = a^n b^m k! / (k-n)! / n!
在模10007的意义下,我们可以计算逆元。
时空复杂度降到O(k)。

#include<cstdio>
#include<cstring>
using namespace std;
#define mod 10007
int rec[1010][1010];    //使用递归组合数必须配个记忆化 ,不加卡成dog
int a,b,k,n,m;
inline int in(){
    int x=0;char ch=getchar();
    while(ch>'9' || ch<'0')    ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
int C(int n,int r){
        if(n==r || r==0)    return rec[n][r]=1;
        if(rec[n][r])    return rec[n][r];
        return rec[n][r]=(C(n-1,r)+C(n-1,r-1))%mod;
    }
    long long power(long long x,int k){    //10006*10006是可能爆 int的 
    long long ans=1;
    while(k){
        if(k&1)    ans=ans*x%mod;
        x=x*x%mod;
        k>>=1;
    }
    return ans;
}
int main(){
    a=in();b=in();k=in();n=in();m=in();
    int ans=power(a,n)*power(b,m)%mod*C(k,m)%mod;
    printf("%d",ans);
    return 0;
}
全部评论

相关推荐

zygg:拼多多挂是不是过一两天就挂的呀
点赞 评论 收藏
分享
小厂面经,也是我的处女面(30min)1.自我介绍2.spring&nbsp;boot的自动装配原理(好多类和接口的单词都忘了全称是啥了,就说了记得的单词,流程应该说对了吧)3.有用过redis吗?主要是用在实现什么功能(说了技术派用redis的zset来实现排行榜)5.有了解过Redisson吗?讲一下对于分布式锁的了解以及在什么场景下应用(说了秒杀场景)6.对mysql有了解吗?包括它的索引优化和创建(把想起来的全说了)7.了解设计模式吗?比如单例模式,为什么要使用单例模式,它的优点是什么(昨天刚看的设计模式)8.工厂模式有了解吗?主要的使用场景是?(也是昨天刚看的)9.场景题:有7个服务器,需要在早上十点定时的向数据库中的用户表中的用户发短信,如果做到发送的消息不重复,且如果发送失败了需要知道是到哪个用户失败了,这样下次就直接从这个用户开始(我答了用spring&nbsp;task来实现定时,用分布式锁来保证只有一份服务器可以发送消息,用消息队列来存储消息,然后用消息确认机制来保证错误信息的记录,以及在数据库或者业务层面完成消息消费的幂等性)10.场景题:如果在系统启动的时间就将数据库的所有用户相关的信息都读到一个hashmap中(这个没啥思路,没答好)27届的投了一个星期终于有一个面试了,大部分公司都只招26的
inari233:已oc,拒了
查看9道真题和解析
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务