传送门
题目:
一开始看题觉得暴力不就好啦:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main(){
ll n,k,ans=0;
cin>>n>>k;
for(ll i=1;i<k&&i<=n;i++){
ans+=k%i;
}
if(n>k)ans+=(k*(n-k));
cout<<ans<<endl;
return 0;
}
结果:TLE……
所以需要优化!
思路:
公式就不证啦,记叭记叭~
注意一些边界条件!!!
代码:
#include <bits/stdc++.h>
#define ll long long
using namespace std;
int main()
{
ll n,k,a;
cin>>n>>k;
ll ans=n*k;
for(ll i=1;i<=n;i=a+1){
if(i>k)break;//i>k所得均为0循环下去无意义啦!(图中有提及)
a=min(k/(k/i),n);//若k/(k/i)>n说明范围超啦,超过那些不计哦!
ans-=(k/i)*(i+a)*(a-i+1)/2;
}
cout<<ans<<endl;
return 0;
}