acdream 1007 a+b 快速幂取模

给出k和n个ai,计算∑ai^k mod 10000000007的值。

#include <iostream>
#include <cstdio>
#define MOD 10000000007
#define ll long long
//模乘法,本质就是用位运算和加法实现除法
//可以防止做乘法时越界
ll mul(ll a,ll b)
{
    ll ans = 0;
    while(b){
        if(b&1) ans = (ans + a) % MOD;
        b = b >> 1;
        a = (a << 1) % MOD;
    }
    ans = ans % MOD;
    return ans;
}
//快速幂取模
ll pow_mod(ll a,ll b)
{
    if(b == 0) return 1;
    ll x = pow_mod(a,b / 2);
    ll ans = mul(x,x)% MOD;
    if(b % 2 == 1) ans = mul(ans,a) % MOD;
    return ans;
}
//
//ll pow_mod(ll a,ll b)
//{
//    ll ans = 1;
//    while(b){
//        if(b&1) ans = mul(ans,a) % MOD;
//        b = b >> 1;
//        a = mul(a,a) % MOD;
//    }
//    return ans;
//}

int main()
{
    int T;
    std::cin>>T;
    //scanf("%lld",&T);
    int n;
    ll m,k;
    while(T--){
        std::cin>>n>>k;
       // scanf("%d%lld",&n,&k);
        ll ans = 0;
        if(k % 2 == 0){
            while(n--){
               // scanf("%lld",&m);
               std::cin>>m;
                if(m < 0) m = m * -1;
                ans = (ans + pow_mod(m,k)) % MOD;
//                if(m == 0 && k == 0){
//                    ans = (ans + 1) % MOD;
//                }
            }
        }else{
            while(n--){
                //scanf("%lld",&m);
                std::cin>>m;
                if(m < 0) {
                    m = m * -1;
                    ans = (ans - pow_mod(m,k)) % MOD;
                }else{
                    ans = (ans + pow_mod(m,k)) % MOD;
                }
            }
        }
        std::cout<<(ans + MOD) % MOD<<std::endl;
        //printf("%lld\n",(ans + MOD) % MOD);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值