蓝桥杯 ALGO-999 数的潜能

试题 算法训练 数的潜能

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　将一个数N分为多个正整数之和，即N=a1+a2+a3+…+ak，定义M=a1a2a3*…*ak为N的潜能。
　　给定N，求它的潜能M。
　　由于M可能过大，只需求M对5218取模的余数。
输入格式
　　输入共一行，为一个正整数N。
输出格式
　　输出共一行，为N的潜能M对5218取模的余数。
样例输入
10
样例输出
36
数据规模和约定
　　1<=N<10^18

3=3 4=2+2，5=2+3，6=3+3，7=3+2+2，8=3+3+2，9=3+3+3
当一个数分解出来的数大于4时又会分解为2*2 所以分到最小的应该是3
对3求余分类讨论 特判n=1的情况

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
typedef long long ll;
ll mod=5218;
ll n,tem,ans;
ll fun(ll a,ll b){
    ll res=1;
    while(b){
        if(b&1)
            res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}
int main(){
    scanf("%lld",&n);
    if(n==1){
        printf("1\n");
        return 0;
    }
    if(n%3==0){
        ans=fun(3,n/3);
    }
    else if(n%3==2){
        ans=2*fun(3,n/3);
    }
    else{
        ans=4*fun(3,n/3-1);
    }
    printf("%lld",ans%mod);
    return 0;
}