ACdream1007 a+b 快速幂求和

题目链接:http://acdream.info/problem?pid=1007


题目大意:给出n个ai和和幂次k,计算∑ai^k mod 10000000007的值。


分析:比较简单的费马小定理应用,注意下ai的值可能是负数就行了,我们可以按k的奇偶来讨论:对于k是偶数,ai的正负对结果是没有影响的,k为奇数时,把ai小于0的情况看做是正数,算出ai^k之后用结果减去这个值即可。


实现代码如下:

#include <cstdio>
#include <iostream>
using namespace std;
typedef long long ll;
const ll mod=10000000007;
ll multi(ll a,ll b)
{
    ll ans=0;
    while(b)
    {
        if(b&1) ans=(ans+a)%mod;
        b>>=1;
        a=(a<<1)%mod;
    }
    ans%=mod;
    return ans;
}
ll quick_mod(ll a,ll k)
{
    ll ans=1;
    while(k)
    {
        if(k&1)
          ans=multi(a,ans)%mod;
        k>>=1;
        a=multi(a,a)%mod;
    }
    return ans;
}
int main()
{
    int t,n;
    ll a,k;
    scanf("%d",&t);
    while(t--)
    {
        ll ans=0;
        scanf("%d%lld",&n,&k);
        if(k&1)
          for(int i=0;i<n;i++)
          {
              scanf("%lld",&a);
              if(a>=0) ans=(ans+quick_mod(a,k))%mod;
              else
              {
                  a*=-1;
                  //a=(a^(a>>63))-(a>>63);
                  ans=(ans-quick_mod(a,k))%mod;
              }
          }
        else
          for(int i=0;i<n;i++)
          {
              scanf("%lld",&a);
              if(a<0) a*=-1;
              //a=(a^(a>>63))-(a>>63);
              ans=(ans+quick_mod(a,k))%mod;
          }
        printf("%lld\n",(ans+mod)%mod);
    }
    return 0;
}


你可能感兴趣的:(ACdream1007 a+b 快速幂求和)