POJ 2800 Joseph's Problem(数论)

201 篇文章 10 订阅

Description
这里写图片描述
Input
两个整数n和k(1<=n,k<=1e9)
Output
输出这里写图片描述
Sample Input
5 3
Sample Output
7
Solution
暴力显然超时,n超过k的部分可以用(n-k)*k得到,所以只要解决[1,k]这个区间即可,首先我们知道当k/i相同时,k%i是公差为k/i的等差数列,那么我们可以从sqrt(k)到2枚举k/i,那么我们就得到了(sqrt(k),n]这个区间上的答案,而求解[1,sqrt(k)]这个区间可以在O(sqrt(k))的时间内解决,这个时间复杂度显然可行
Code

#include<cstdio>
#include<iostream>
#include<cmath>
using namespace std;
typedef long long ll;
ll solve(ll n,ll k)
{
    ll p=(ll)sqrt(1.0*k),q=k/p,ans=0;
    if(n>k)ans+=(n-k)*k;//大于k的部分 
    for(ll i=p;i>1;i--)
    {
        ll s=k/i,e=k/(i-1);
        if(s>n)break;//s大于n则退出循环 
        if(e>n)e=n;//e于n则令其为n只求s~n这一段 
        ans+=(k%(s+1)+k%e)*(e-s)/2;//等差数列求和公式 
    }
    for(ll i=1;i<=n&&i<=q;i++)//求[1,sqrt(k)]部分 
        ans+=k%i;
    return ans;
}
int main()
{
    ll n,k;
    while(~scanf("%lld%lld",&n,&k))
        printf("%lld",solve(n,k));
    return 0;
} 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值