P2261 余数求和(整除分块)

最新推荐文章于 2024-04-01 16:35:43 发布

sedcftyv

最新推荐文章于 2024-04-01 16:35:43 发布

阅读量151

点赞数

分类专栏：数论

本文链接：https://blog.csdn.net/sedcftyv/article/details/102648714

版权

数论专栏收录该内容

16 篇文章 0 订阅

订阅专栏

$\sum^{n}_{i=1}(k\ mod\ i)=\sum _{i=1}^{n}(k-\left \lfloor \frac{k}{i} \right \rfloor\times i)=n\times k- \sum^{n}_{i=1}(i\times \left \lfloor \frac{k}{i} \right \rfloor)\;\;\;O(2\sqrt{k})$

1.其中i从1增加到n为等差数列， $\left \lfloor \frac{k}{i} \right \rfloor$ 的值随i的增加具有阶梯性(下降)，故可将其分为一个一个台阶进行计算。即整除分块

对于一个 $\left \lfloor \frac{k}{i_{1}} \right \rfloor$ ,值与其相同的最后一个 $\left \lfloor \frac{k}{i_{2}} \right \rfloor$ 有 $i_{2}=\left \lfloor \frac{k}{\left \lfloor \frac{k}{i_{1}} \right \rfloor} \right \rfloor$ ，利用该性质对每个台阶左端点 $i_{1}$ 可求出台阶右端点 $i_{2}$ 。

2.对于每个台阶， $\left \lfloor \frac{k}{i} \right \rfloor$ 相同，i的值为等差数列，设l,r分别为台阶左、右端点。每个台阶的值： $\sum^{l}_{i=r}(i\times \left \lfloor \frac{k}{i} \right \rfloor)= \left \lfloor \frac{k}{i} \right \rfloor \times \sum_{i=r}^{l}i= \left \lfloor \frac{k}{i} \right \rfloor \times \frac{(l+r)\times(r-l+1)}{2}$

3.ans=n*k-所有台阶的值之和。

注意判断 $\frac{k}{i}=0$ 的情况，以及台阶右端点不能大于n。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
int main()
{
    //freopen("1.txt","r",stdin);
    ll n,k,ans;
    scanf("%lld%lld",&n,&k);
    ll l,r;
    ans=n*k;
    for(ll i=1;i<=n;)
    {
    if(k/i==0)
        break;
    r=min(k/(k/i),n);
    ans-=(i+r)*(r-i+1)/2*(k/i);
        i=r+1;
  //  cout<<1<<endl;
    }
    printf("%lld\n",ans);
    return 0;
}