7-3 2020 CCPC Wannafly Winter Camp Day1 C. 染色图

定义一张无向图 G=⟨V,E⟩ 是 k 可染色的当且仅当存在函数 f:V↦{1,2,⋯,k} 满足对于 G 中的任何一条边 (u,v)，都有 f(u)≠f(v)。

定义函数 g(n,k) 的值为所有包含 n 个点的无自环、无重边的 k 可染色无向图中的边数最大值。举例来说，g(3,1)=0,g(3,2)=2,g(3,3)=3。

现在给出三个整数 n,l,r，你需要求解： (∑​​​ g(n,i))mod 998244353 from l to r)
输入格式：
第一行输入一个整数 T(1≤T≤10​³​​ )，表示数据组数。

对于每组数据，输入三个整数 n,l,r(1≤l≤r≤n≤10^​9​ )。

输出格式
对于每组数据，输出一行一个整数表示答案。

输入样例：
5
3 1 1
3 2 2
5 2 4
10 3 9
1000 123 789

输出样例：
0
2
23
280
332539617

题解：把图分成几个完全图，每个完全图内有1–k种颜色（每种颜色一个节点），各个完全图再相互连接即可得最优解。
数论分块即可求得。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<cstdlib>
#include<map>
#include<set>
#define ll long long
#define llu unsigned ll
#define ld long double
#define pr make_pair
#define pb push_back
#define x first
#define y second
#define int ll
using namespace std;
const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const int mod=998244353;
int n,nl,nr;
int inv2;

ll mypow(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}

signed main(void)
{
    int t;
    scanf("%lld",&t);
    inv2=mypow(2,mod-2);
    while(t--)
    {
        scanf("%lld%lld%lld",&n,&nl,&nr);
        int ans=(n*n%mod-n+mod)%mod*(nr-nl+1)%mod;
        ans%=mod;
        for(int l=nl,r;l<=nr;l=r+1)
        {
            r=min(n/(n/l),nr);
            int now=n/l;
            ans=(ans+(now*now%mod+now)*(r-l+1)%mod*(r+l)%mod*inv2%mod)%mod;
            ans=(ans-2*n*now%mod*(r-l+1)%mod+mod)%mod;
        }
        printf("%lld\n",ans%mod*inv2%mod);
    }
    return 0;
}