洛谷P2261 [CQOI2007]余数求和(整除分块)+一点学习笔记

题目背景

数学题，无背景

题目描述

给出正整数n和k，计算G(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n的值，其中k mod i表示k除以i的余数。例如G(10, 5)=5 mod 1 + 5 mod 2 + 5 mod 3 + 5 mod 4 + 5 mod 5 …… + 5 mod 10=0+1+2+1+0+5+5+5+5+5=29

输入输出格式

输入格式：

 

两个整数n k

 

输出格式：

 

答案

 

输入输出样例

输入样例#1： 复制

10 5

输出样例#1： 复制

29

说明

30%: n,k <= 1000

60%: n,k <= 10^6

100% n,k <= 10^9

题解:题意很简单,对于1e9范围内的余数求和,线性复杂度肯定是不够的,这个时候就需要一个很神奇的思想,整除分块:~~

首先考虑对于(n/i)从1-n求和,因为n/i很多的区间是相同的 从l-r中n/l==n/r,这些地方便可以直接跳过去大大的减小复杂度,

代码大概是这样的:

for(int l=1,r;l<=n;l=r+1)
    {
        r=n/(n/l);
        ans+=(r-l+1)*(n/l);
    }

这样就可以只有sqrt(n)的复杂度了~~

那么我们回到这道题,对于这道题来说k%i=k-(k/i)*i; 那么便可以发现在一个区间l-r中,k/i是相同的,那么便可以直接求i的一个等差数列的和即可~~

#include<bits/stdc++.h>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define SI(i) scanf("%lld",&i)
#define PI(i) printf("%lld\n",i)
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int MAX=1e5+5;
const int INF=0x3f3f3f3f;
const double eps=1e-8;
int dir[9][2]={0,1,0,-1,1,0,-1,0, -1,-1,-1,1,1,-1,1,1};
template<class T>bool gmax(T &a,T b){return a<b?a=b,1:0;}
template<class T>bool gmin(T &a,T b){return a>b?a=b,1:0;}
template<class T>void gmod(T &a,T b){a=((a+b)%mod+mod)%mod;}
typedef pair<ll,ll> PII;

int main()
{
    ll n,k;
    SI(n);SI(k);

    ll ans=n*k;
    for(ll l=1,r;l<=min(n,k);l=r+1)
    {
        r=min(k/(k/l),n);
        ans-=(k/l)*(l+r)*(r-l+1)/2;
    }

    printf("%lld\n",ans);

    return 0;
}