【数论】CF1789E Serval and Music Game 题解

Serval and Music Game

题面翻译

题目描述

给定整数 $n$ 和长度为 $n$ 的递增序列 $s$。
定义 $f(x)$ 为满足下列要求的整数 $i(1\le i\le n)$ 的数量：

• 存在非负整数 $p_i,q_i$ 使得 $s_i=p_i\lfloor \frac{s_n}{x}\rfloor +q_i\lceil \frac{s_n}{x}\rceil$。

你需要求出 ${\sum }_{x=1}^{s_n}x\times f(x)$ 对 $998244353$ 取模后的值。
每个测试点包含 $t$ 组数据。

输入格式

第一行输入一个整数 $t(1\le t\le 10^4)$ 表示数据组数，接下来对于每组数据：
第一行输入一个整数 $n(1\le n\le 10^6)$。
接下来输入一行 $n$ 个整数表示序列 $s(1\le s_1<s_2<\cdots <s_n\le 10^7)$。
单个测试点内所有组数据对应的 $n$ 之和不超过 $10^6$，对应的 $s_n$ 之和不超过 $10^7$。

输出格式

对于每组数据：
输出一行一个整数表示 ${\sum }_{x=1}^{s_n}x\times f(x)$ 对 $998244353$ 取模后的值。

题目描述

Serval loves playing music games. He meets a problem when playing music games, and he leaves it for you to solve.

You are given $n$ positive integers $s_1<s_2<\dots <s_n$ . $f(x)$ is defined as the number of $i$ ( $1\le i\le n$ ) that exist non-negative integers $p_i,q_i$ such that:

$$s_i=p_i\lfloor \frac{s_n}{x}\rfloor +q_i\lceil \frac{s_n}{x}\rceil$$

Find out ${\sum }_{x=1}^{s_n}x\cdot f(x)$ modulo $998,244,353 $ .

As a reminder, $\lfloor x\rfloor$ denotes the maximal integer that is no greater than $x$ , and $\lceil x\rceil$ denotes the minimal integer that is no less than $x$.

输入格式

Each test contains multiple test cases. The first line contains the number of test cases $t$ ( $1\le t\le 10^4$ ). The description of the test cases follows.

The first line of each test cases contains a single integer $n$ ( $1\le n\le 10^6$ ).

The second line of each test case contains $n$ positive integers $s_1,s_2,\dots ,s_n$ ( $1\le s_1<s_2<\dots <s_n\le 10^7$ ).

It is guaranteed that the sum of $n$ over all test cases does not exceed $10^6$ , and the sum of $s_n$ does not exceed $10^7$ .

输出格式

For each test case, print a single integer in a single line — the sum of $x\cdot f(x)$ over all possible $x$ modulo $998244353$ .

样例 #1

样例输入 #1

4
3
1 2 4
4
1 2 7 9
4
344208 591000 4779956 5403429
5
1633 1661 1741 2134 2221

样例输出 #1

26
158
758737625
12334970

提示

Solution

考虑每个 $x$ 对答案的贡献。

分类讨论：

若 $x|s_n$ ，则可以枚举 $s_n/x$ 的倍数计算满足要求的 $s_i$ 的个数。

若 $x\nmid s_n$ ，首先有 $\lceil \frac{s_n}{x}\rceil =\lfloor \frac{s_n}{x}\rfloor +1$

设 $\lfloor \frac{s_n}{x}\rfloor =k$ ，则 $s_i=(p_i+q_i)k+q_i$ ，由于 $p_i,q_i\in \mathbb{N}$ ，

则若 $s_i$ 满足要求，必然有 $\lfloor \frac{s_i}{k}\rfloor \ge q_i$ ，不难证明 $q_i=s_i\mathrm{mod}k$ ，

此时枚举 $j=\lfloor \frac{s_i}{k}\rfloor$ ，那么 $s_i$ 满足要求的充要条件为 $s_i\in [jk,jk+j]$ 。

可以发现当 $0\le q_i<k$ ，故当 $j\ge k$ 时，此后的 $s_i$ 都会满足要求。

因此我们只需枚举 $j<k$ ，剩下的统一处理（见代码）。

对于一个区间内满足要求的 $s_i$ ，可以用一个桶存储出现次数并将其进行前缀和，记为 $cnt$ 数组，

那么在 $[l,r]$ 中的 $s_i$ 的个数就是 $cn{t}_r-cn{t}_{l-1}$ 。

这样我们可以 $O(\sqrt{s_n})$ 求出 $f(x)$ 的值。

考虑到 $k=\lfloor \frac{s_n}{x}\rfloor$ 可以数论分块，于是可以在 $O(s_n)$ 求出答案。

Code

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N = 1e7 + 3;
const LL MOD = 998244353;

inline void read(int &x)
{
    int sgn = 1; x = 0;
    char ch = getchar();
    while(ch != '-' && (ch < '0' || ch > '9')) ch = getchar();
    if(ch == '-') sgn = -1, ch = getchar();
    while(ch >= '0' && ch <= '9') x = (x << 1) + (x << 3), x += (ch ^ '0'), ch = getchar();
    x *= sgn;
}

int T, cnt[N], n, s[N], m;

int main()
{
    read(T);
    while(T -- )
    {
        read(n); m = 0;
        for(int i = 1; i <= n; i ++ ) read(s[i]), m = max(m, s[i]);
        for(int i = 0; i <= m; i ++ ) cnt[i] = 0;
        for(int i = 1; i <= n; i ++ ) cnt[s[i]] ++ ;
        for(int i = 1; i <= m; i ++ ) cnt[i] += cnt[i - 1];
        LL res = 0, tp = 0;
        for(int i = 1; i <= m; i ++ )
        {
            LL sum = 0; int k = m / i;
            if(m % i)
                if(m % (i - 1) && m / (i - 1) == k) sum = tp;
                else
                {
                    for(int j = 1; j < k && j * k <= m; j ++ ) sum += cnt[min(j * k + j, m)] - cnt[j * k - 1];
                    if(1ll * k * k <= 1ll * m) sum += cnt[m] - cnt[k * k - 1];
                }  
            else for(int j = k; j <= m; j += k) sum += cnt[j] - cnt[j - 1];
            res += i * sum % MOD; res %= MOD;
            tp = sum; 
        }
        printf("%lld\n", res);
    }
    return 0;
}