兰州大学第一届『飞马杯』糖果魔法阵（扩展欧拉定理+光速幂+打表找规律）

最新推荐文章于 2021-06-04 00:11:01 发布

Sankkl1

最新推荐文章于 2021-06-04 00:11:01 发布

阅读量254

点赞数 1

分类专栏：牛客文章标签：数学 c++ c语言

本文链接：https://blog.csdn.net/Sankkl1/article/details/117427408

版权

牛客专栏收录该内容

18 篇文章 0 订阅

订阅专栏

题目链接：https://ac.nowcoder.com/acm/contest/16520/G

分析

看了pdf题解，感觉上面推的公式~~太难了~~太过复杂，可以通过打表的方式找出规律。

在这里插入图片描述
对于这个更新糖果数的公式，我们写出它的前几项：

不难发现，在第10项之后它开始了4项为一个周期的循环，所以在我们知道已经使用了 k 次魔法阵的情况下可以直接得到这时的糖果数目。

然后就是下面这个已知天数和糖果数更新能量的式子：
在这里插入图片描述
根据扩展欧拉定理：

于是我们可以对指数进行取余
$4^{k} \equiv 4^{k\mod\varphi(998244353)} (\mod 998244353)$
$\equiv n\mod\varphi(998244352)+\varphi(998244352)(n \geq \varphi(998244352))(\mod 998244353)$

如果用快速幂的话会超时。因为底数固定，所以我们可以用光速幂来计算。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define maxn 35000
#define fi first
#define se second
#define pb push_back

const ll mod = 998244353, phi = 402653184;
ll n,s,C,q,S;
ll cd[15]={1,915057325,313715344,828907141,687199838,216660350,397265547,16066133,150000439,661125904,521509362,713551262,678018512,284693091,320225841};
ll a[maxn+5][2],b[maxn+5][2],c[maxn+5][2];

void Init()
{
	a[0][0] = a[0][1] = b[0][0] = b[0][1] = c[0][0] = c[0][1] = 1;
	for(int i=1;i<=maxn;i++)
	{
		a[i][0] = a[i - 1][0] * 116195172 % mod;
		b[i][0] = b[i - 1][0] * 882049183 % mod;
		c[i][0] = c[i - 1][0] * 4 % (mod - 1);
	}
	for(int i=1;i<=maxn;i++)
	{
		a[i][1] = a[i - 1][1] * a[maxn][0] % mod;
		b[i][1] = b[i - 1][1] * b[maxn][0] % mod;
		c[i][1] = c[i - 1][1] * c[maxn][0] % (mod - 1);
	}
}

int euler_phi(int n) {
    int ans = n;
    for (int i = 2; i * i <= n; i++)
        if (n % i == 0) {
            ans = ans / i * (i - 1);
            while (n % i == 0) n /= i;
        }
    if (n > 1) ans = ans / n * (n - 1);
    return ans;
}

ll cal(ll tmp)
{
    if(tmp > 14)
    {
        tmp -= 14;
        if(tmp % 4 == 0) tmp = 14;
        else tmp = tmp % 4 + 10;
    }
    return cd[tmp];
}

int main()
{
    //cout<<euler_phi(mod - 1);
    Init();
    int T = 1;
    //scanf("%d",&T);
    while(T--)
    {
        ll k;
        
        scanf("%lld%lld%lld",&S,&C,&q);
        while(q--)
        {
            k = 0;
            scanf("%lld",&n);
            s = S;
            for(ll i=1;i<=n;i++)
            {
                k += s / C;
                s %= C;
                if(k >= phi) k = k % phi + phi;
			    ll tmp = c[k / maxn][1] * c[k % maxn][0] % (mod - 1);
			    s = (a[tmp / maxn][1] * a[tmp % maxn][0] % mod - b[tmp / maxn][1] * b[tmp % maxn][0] % mod + mod) % mod * i % mod * cal(k) % mod;
            }
            printf("%lld\n",cal(k));
        }
    }
    return 0;
}