题目背景
数学题,无背景
题目描述
给出正整数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;
}