Loading [MathJax]/extensions/TeX/mathchoice.js

loj6485. LJJ 学二项式定理

题意

给出n,s,a0,a1,a2,a3,求

\sum_{i = 0} ^ n \binom{n}{i} s ^ i a_{i \mod 4}

998244353取模的值。

题解

本来像强行用xiaomange的方法艹过去,but failed。
大概猜到和原根有关,但是思考也只是停留在特殊的情况下,没有进行更有效的思考。
正解是单位根反演。
大概就是在推式子的过程中发现突破口的

\begin{aligned} origin & = \sum_{k = 0} ^ 3 a_k \sum_{i = 0} ^ n [4 | i + 4 - k] \binom{n}{i} s ^ i \\ \end{aligned}

注意到在对998244353取模下有4次单位根\omega_4,又由于有恒等式

[n | m] = \frac{1}{n} \sum_{i = 0} ^ {n - 1} \omega_n ^ {im}

\begin{aligned} origin & = \sum_{k = 0} ^ 3 a_k \sum_{i = 0} ^ n \binom{n}{i} s ^ i \sum_{j = 0} ^ 3 \omega_4 ^ {j(i + 4 - k)} \\ & = \sum_{k = 0} ^ 3 a_k \sum_{j = 0} ^ 3 \sum_{i = 0} ^ n \binom{n}{i} s ^ i \omega_4 ^ {j(i + 4 - k)} \\ & = \sum_{k = 0} ^ 3 a_k \sum_{j = 0} ^ 3 \omega_4 ^ {j(4 - k)} \sum_{i = 0} ^ n \binom{n}{i} s ^ i \omega_4 ^ {ji} \\ & = \sum_{k = 0} ^ 3 a_k \sum_{j = 0} ^ 3 \omega_4 ^ {j(4 - k)} (s \omega_4 ^ j) ^ n \\ \end{aligned}

再直接做即可。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 998244353, g = 3;
ll power (ll a, ll b) {
	ll ret = 1;
	for ( ; b; b >>= 1, a = a * a % mod) {
		if (b & 1) {
			ret = ret * a % mod;
		}
	}
	return ret;
}
int T; ll w, n, s, a, p, q, ans;
int main () {
	w = power(g, (mod - 1) / 4);
	for (scanf("%d", &T); T; --T) {
		scanf("%lld%lld", &n, &s), ans = 0;
        for (int i = 0; i < 4; ++i) {
        	scanf("%lld", &a), p = s, q = 1;
        	for (int j = 0; j < 4; ++j) {
        		ans += a * power(q, 4 - i) % mod * power(p + 1, n);
        		ans %= mod;
        		p = p * w % mod;
        		q = q * w % mod;
        	}
        }
        ans = ans * power(4, mod - 2) % mod;
        printf("%lld\n", ans);
    }
    return 0;
}
posted @   psimonw  阅读(205)  评论(0编辑  收藏  举报
编辑推荐:
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
阅读排行:
· 手把手教你在本地部署DeepSeek R1,搭建web-ui ,建议收藏!
· Spring AI + Ollama 实现 deepseek-r1 的API服务和调用
· 数据库服务器 SQL Server 版本升级公告
· C#/.NET/.NET Core技术前沿周刊 | 第 23 期(2025年1.20-1.26)
· 程序员常用高效实用工具推荐,办公效率提升利器!
点击右上角即可分享
微信分享提示