\(\mathcal{Description}\)

  Link.

  \(T\) 组询问,每次给出 \(n,a,b,c,k_1,k_2\),求

\[\sum_{x=0}^nx^{k_1}\left\lfloor\frac{ax+b}{c}\right\rfloor^{k_2}\bmod(10^9+7)
\]

  \(T=1000\),\(n,a,b,c\le10^9\),\(0\le k_1+k_2\le 10\)。

\(\mathcal{Solution}\)

  类欧模板题的集大成者。

  令 \(f(n,a,b,c,k_1,k_2)\) 表示所求答案,尝试通过分类讨论递归到不同的情况求解。

  1. \(a=0\lor an+b<c\)

    \[f(n,a,b,c,k_1,k_2)=\left\lfloor\frac{an+b}{c}\right\rfloor^{k_2}\sum_{x=0}^nx^{k_1}
    \]

    预处理 \(k+1\) 次多项式 \(g_k(x)=\sum_{i=0}^xi^k\) 的各项系数,这个就能直接算出来。

  2. 否则若 \(a\ge c\)

    \[\begin{aligned}
    f(n,a,b,c,k_1,k_2)&=\sum_{x=0}^nx^{k_1} \left(\left\lfloor\frac{a}{c}\right\rfloor x+\left\lfloor\frac{(a\bmod c)x+b}{c}\right\rfloor \right)^{k_2}\\
    &=\sum_{i=0}^{k_2} \binom{k_2}{i} \left\lfloor\frac{a}{c}\right\rfloor^i\sum_{x=0}^n x^{k_1+i}\left\lfloor\frac{(a\bmod c)x+b}{c}\right\rfloor^{k_2-i}\\
    &=\sum_{i=0}^{k_2} \binom{k_2}{i} \left\lfloor\frac{a}{c}\right\rfloor^i f(n,a\bmod c,b,c,k_1+i,k_2-i)
    \end{aligned}
    \]

    递归处理。

  3. 否则若 \(b\ge c\)

    类似 2.,最终得到

    \[f(n,a,b,c,k_1,k_2)=\sum_{i=0}^{k_2} \binom{k_2}{i} \left\lfloor\frac{b}{c}\right\rfloor^i f(n,a,b\bmod c,c,k_1,k_2-i)
    \]

    递归处理。

  4. 对于其余情况

    考虑把高斯函数丢到求和指标上,注意到

    \[m^k=\sum_{j=0}^{m-1}[(j+1)^k-j^k]
    \]

    以此替代 \(\left\lfloor\frac{ax+b}{c}\right\rfloor^{k_2}\),并交换求和顺序,得到

    \[\begin{aligned}
    f(n,a,b,c,k_1,k_2)&=\sum_{j=0}^{\left\lfloor\frac{an+b}{c}\right\rfloor-1}[(j+1)^{k_2}-j^{k_2}]\sum_{x=0}^nx^{k_1}\left[x>\left\lfloor\frac{cj+c-b-1}{a}\right\rfloor \right]\\
    &= \sum_{j=0}^{\left\lfloor\frac{an+b}{c}\right\rfloor-1} [(j+1)^{k_2}-j^{k_2}]\sum_{x=0}^n x^{k_1}-\sum_{j=0}^{\left\lfloor\frac{an+b}{c}\right\rfloor-1} [(j+1)^{k_2}-j^{k_2}]\sum_{x=0}^{\left\lfloor\frac{cj+c-b-1}{a}\right\rfloor} x^{k_1}
    \end{aligned}
    \]

    前半部分直接计算,研究后半部分,发现最后一个求和符号是 \(g_{k_1}\left(\left\lfloor\frac{cj+c-b-1}{a}\right\rfloor\right)\),枚举次数将其展开:

    \[\begin{aligned}
    &=\sum_{i=0}^{k_2-1}\binom{k_2}{i} \sum_{j=0}^{k_1+1}([x^j]g_{k_1})\sum_{x=0}^{\left\lfloor\frac{an+b}{c}\right\rfloor-1}x^i\left\lfloor\frac{cx+c-b-1}{a}\right\rfloor^j\\
    &=\sum_{i=0}^{k_2-1}\binom{k_2}{i} \sum_{j=0}^{k_1+1}([x^j]g_{k_1})f\left(\left\lfloor\frac{an+b}{c}\right\rfloor-1,c,c-b-1,a,i,j\right)
    \end{aligned}
    \]

    其中 \(i\) 枚举 \((j+1)^{k_2}\) 的次数;\(j\) 枚举 \(g_{k_1}\) 的次数;\(x\) 枚举原来的 \(j\)。这样也能递归求解了。

  递归中指标变换形如欧几里得算法,设指标值域为 \(V\),\(K=k_1+k_2\),则复杂度为 \(\mathcal O(TK^4\log V)\)。

\(\mathcal{Code}\)

  实现时,可以对于 \(n,a,b,c\),一次算出所有可能的 \(f(n,a,b,c,k_1,k_2)\) 的值。

/*~Rainybunny~*/

#include <cstdio>
#include <cassert>
#include <iostream> #define rep( i, l, r ) for ( int i = l, rep##i = r; i <= rep##i; ++i )
#define per( i, r, l ) for ( int i = r, per##i = l; i >= per##i; --i ) typedef long long LL; const int MOD = 1e9 + 7;
int comb[15][15]; inline void subeq( int& a, const int b ) { ( a -= b ) < 0 && ( a += MOD ); }
inline int sub( int a, const int b ) { return ( a -= b ) < 0 ? a + MOD : a; }
inline int mul( const long long a, const int b ) { return int( a * b % MOD ); }
inline int add( int a, const int b ) { return ( a += b ) < MOD ? a : a - MOD; }
inline void addeq( int& a, const int b ) { ( a += b ) >= MOD && ( a -= MOD ); }
inline int mpow( int a, int b ) {
int ret = 1;
for ( ; b; a = mul( a, a ), b >>= 1 ) ret = mul( ret, b & 1 ? a : 1 );
return ret;
} struct PowerPoly {
int n, a[12];
PowerPoly() {} PowerPoly( const int k ): n( k ) {
static int mat[15][15];
for ( int i = 0, p = 0; i <= n + 1; ++i ) {
addeq( p, mpow( i, n ) );
mat[i][n + 2] = p;
}
rep ( i, 0, n + 1 ) {
for ( int j = 0, p = 1; j <= n + 1; ++j, p = mul( p, i ) ) {
mat[i][j] = p;
}
} rep ( i, 0, n + 1 ) {
int p = -1;
rep ( j, i, n + 1 ) if ( mat[j][i] ) { p = j; break; }
assert( ~p );
if ( p != i ) std::swap( mat[i], mat[p] );
int iv = mpow( mat[i][i], MOD - 2 );
rep ( j, i + 1, n + 1 ) {
int t = mul( iv, mat[j][i] );
rep ( l, i, n + 2 ) subeq( mat[j][l], mul( mat[i][l], t ) );
}
}
per ( i, n + 1, 0 ) {
a[i] = mul( mat[i][n + 2], mpow( mat[i][i], MOD - 2 ) );
rep ( j, 0, i - 1 ) subeq( mat[j][n + 2], mul( a[i], mat[j][i] ) );
}
} inline int operator () ( LL x ) const {
int ret = 0; x %= MOD;
for ( int i = 0, p = 1; i <= n + 1; ++i, p = mul( x, p ) ) {
addeq( ret, mul( a[i], p ) );
}
return ret;
}
}; const PowerPoly pwr[11]
= { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }; struct Atom {
int res[11][11];
Atom(): res{} {}
inline int* operator [] ( const int k ) { return res[k]; }
}; /* *
* calculate \sum_{x=0}^n x^{k_1} \lfloor \frac{ax+b}{c} \rfloor^{k_2},
* where k_1+k_2 <= 10.
* */
inline Atom solve( const LL n, const LL a, const LL b, const LL c ) {
Atom ret; LL m = ( a * n + b ) / c;
if ( !a || 1ll * a * n + b < c ) {
rep ( k1, 0, 10 ) {
int t = int( m % MOD ), p = 1;
rep ( k2, 0, 10 - k1 ) {
ret[k1][k2] = mul( p, pwr[k1]( n ) );
p = mul( p, t );
}
}
return ret;
} if ( a >= c ) {
Atom tmp( solve( n, a % c, b, c ) );
rep ( k1, 0, 10 ) rep ( k2, 0, 10 - k1 ) {
int t = int( ( a / c ) % MOD ), p = 1;
rep ( i, 0, k2 ) {
addeq( ret[k1][k2], mul( comb[k2][i],
mul( p, tmp[k1 + i][k2 - i] ) ) );
p = mul( p, t );
}
}
return ret;
} if ( b >= c ) {
Atom tmp( solve( n, a, b % c, c ) );
rep ( k1, 0, 10 ) rep ( k2, 0, 10 ) {
int t = int( ( b / c ) % MOD ), p = 1;
rep ( i, 0, k2 ) {
addeq( ret[k1][k2], mul( comb[k2][i],
mul( p, tmp[k1][k2 - i] ) ) );
p = mul( p, t );
}
}
return ret;
} Atom tmp( solve( m - 1, c, c - b - 1, a ) );
rep ( k1, 0, 10 ) {
int t = int( m % MOD ), p = 1;
rep ( k2, 0, 10 - k1 ) {
ret[k1][k2] = mul( p, pwr[k1]( n ) );
rep ( i, 0, k2 - 1 ) rep ( j, 0, k1 + 1 ) {
subeq( ret[k1][k2], mul( comb[k2][i],
mul( pwr[k1].a[j], tmp[i][j] ) ) );
}
p = mul( t, p );
}
}
return ret;
} int main() {
comb[0][0] = 1;
rep ( i, 1, 10 ) {
comb[i][0] = 1;
rep ( j, 1, i ) comb[i][j] = add( comb[i - 1][j - 1], comb[i - 1][j] );
} int n, a, b, c, k1, k2, T;
for ( scanf( "%d", &T ); T--; ) {
scanf( "%d %d %d %d %d %d", &n, &a, &b, &c, &k1, &k2 );
printf( "%d\n", solve( n, a, b, c )[k1][k2] );
}
return 0;
}

Solution -「LOJ #138」「模板」类欧几里得算法的更多相关文章

  1. Luogu 5170 【模板】类欧几里得算法

    原理不难但是写起来非常复杂的东西. 我觉得讲得非常好懂的博客.   传送门 我们设 $$f(a, b, c, n) = \sum_{i = 0}^{n}\left \lfloor \frac{ai + ...

  2. 洛谷P5170 【模板】类欧几里得算法(数论)

    传送门 此题剧毒,公式恐惧症患者请直接转去代码→_→ 前置芝士 基本数论芝士 题解 本题就是要我们求三个函数的值 \[f(a,b,c,n)=\sum_{i=0}^n \left\lfloor\frac ...

  3. 「LOJ 556 Antileaf's Round」咱们去烧菜吧

    「LOJ 556 Antileaf's Round」咱们去烧菜吧 最近在看 jcvb 的生成函数课件,顺便切一切上面讲到的内容的板子题,这个题和课件上举例的背包计数基本一样. 解题思路 首先列出答案的 ...

  4. Solution -「Luogu 5170」类欧几里得算法

    推柿子大赛了属于是. 题目要求三个柿子,不妨分别记为: \[\begin {align} f (a, b, c, n) &= \sum \limits _{i = 0} ^{n} \lfloo ...

  5. Note -「多项式」基础模板(FFT/NTT/多模 NTT)光速入门

      进阶篇戳这里. 目录 何为「多项式」 基本概念 系数表示法 & 点值表示法 傅里叶(Fourier)变换 概述 前置知识 - 复数 单位根 快速傅里叶正变换(FFT) 快速傅里叶逆变换(I ...

  6. 【LOJ#6066】「2017 山东一轮集训 Day3」第二题(哈希,二分)

    [LOJ#6066]「2017 山东一轮集训 Day3」第二题(哈希,二分) 题面 LOJ 题解 要哈希是很显然的,那么就考虑哈希什么... 要找一个东西可以表示一棵树,所以我们找到了括号序列. 那么 ...

  7. [LOJ 6031]「雅礼集训 2017 Day1」字符串

    [LOJ 6031] 「雅礼集训 2017 Day1」字符串 题意 给定一个长度为 \(n\) 的字符串 \(s\), \(m\) 对 \((l_i,r_i)\), 回答 \(q\) 个询问. 每个询 ...

  8. [LOJ 6030]「雅礼集训 2017 Day1」矩阵

    [LOJ 6030] 「雅礼集训 2017 Day1」矩阵 题意 给定一个 \(n\times n\) 的 01 矩阵, 每次操作可以将一行转置后赋值给某一列, 问最少几次操作能让矩阵全为 1. 无解 ...

  9. [LOJ 6029]「雅礼集训 2017 Day1」市场

    [LOJ 6029] 「雅礼集训 2017 Day1」市场 题意 给定一个长度为 \(n\) 的数列(从 \(0\) 开始标号), 要求执行 \(q\) 次操作, 每次操作为如下四种操作之一: 1 l ...

随机推荐

  1. js 关于replace() 的使用心得

    1.前言 我想把一段话 let a = "抱歉,您当前的主治医生有紧急情况不得不下班,您的预约将由<br>医生:里斯<br>为您就诊,<br>诊室位置:门 ...

  2. Vulnhub系列:Tomato(文件包含getshell)

    这个靶机挺有意思,它是通过文件包含漏洞进行的getshell,主要姿势是将含有一句话木马的内容记录到ssh的登录日志中,然后利用文件包含漏洞进行包含,从而拿到shell 0x01 靶机信息 靶机:To ...

  3. Sentry 开发者贡献指南 - Django Rest Framework(Serializers)

    Serializer 用于获取复杂的 python 模型并将它们转换为 json.序列化程序还可用于在验证传入数据后将 json 反序列化回 Python 模型. 在 Sentry,我们有两种不同类型 ...

  4. Android官方文档翻译 一 Getting Started

    Getting Started 让我们开始吧! Welcome to Training for Android developers. 欢迎来到Android开发者训练营. Here you'll f ...

  5. Spring 官宣发布 Spring Boot 3.0 第一个里程碑 M1,从 Java 8 提升到 Java 17!

    Spring官方于2022年1月20日发布Spring Boot 3.0.0-M1版本,预示开启了Spring Boot 3.0的里程碑,相信这是通往下一代Spring框架的激动人心的旅程. 接下来一 ...

  6. Java中四种访问权限总结

    一.Java中有四种访问权限, 其中三种有访问权限修饰符,分别为private.public.protected,还有一种不带任何修饰符(default). 1. private: Java语言中对访 ...

  7. 不难懂——CSS 匹配指定name元素

    <!doctype html> <html> <head> <meta charset="utf-8"> <meta name ...

  8. 字节跳动 iOS Heimdallr 卡死卡顿监控方案与优化之路

    点这里申请 本文主要介绍Heimdallr对卡死.卡顿异常的监控原理,并结合长时间的业务沉淀发现的问题进行不断迭代和优化,逐步实现全面.稳定.可靠的历程. 作者:字节跳动终端技术--白昆仑 前言 卡死 ...

  9. Power Apps 创建响应式布局

    前言 我们都知道Power Apps作为低代码平台,最大的优势就是各个设备之间的兼容性,尤其是自带的响应式布局,非常好用. 这不,我们就为大家分享一下,如何使用Power Apps画布应用,创建响应式 ...

  10. Iceberg学习日记(1) 定位两个线上Iceberg查不到文件的问题

    前言 Iceberg是我们去年年底(2020)开始调研,目前上线了130多张表.主要用于流量日志清洗,数据报表,推荐特征基础数据.至今为也算是积累了一些使用及定位问题经验. 这篇文章会介绍两个线上Ic ...