Joseph's Problem UVALive - 3521（等差数列的应用）

题意：给定n, k,求出∑ni=1(k mod i)

思路：由于n和k都很大，直接暴力是行不通的，然后在纸上画了一些情况，就发现其实对于k/i相同的那些项是形成等差数列的，于是就可以把整个序列进行拆分成[k,k/2],[k/2, k/3], [k/3,k/4]...k[k/a, k/b]这样的等差数列，利用大步小步算法思想，这里a枚举到sqrt(k)就可以了，这样就还剩下[1,k/a]的序列需要去枚举，总时间复杂度为O(sqrt(k))，然后注意对于n大于k的情况，n超过k的部分全是等于k，为(n - k) * k,这样把所有部分加起来就是答案

转载至：https://blog.csdn.net/accelerator_/article/details/36949761

#include <iostream>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <cmath>
#define MOD 2018
#define LL long long
#define ULL unsigned long long
#define Pair pair<int, int>
#define mem(a, b) memset(a, b, sizeof(a))
#define _  ios_base::sync_with_stdio(0),cin.tie(0)
//freopen("1.txt", "r", stdin);
using namespace std;
const int maxn = , INF = 0x7fffffff;

int main()
{
    LL k, n;
    while(cin>> n >> k)
    {
        LL res = ;
        if(n > k) res = (n-k) * k;
        LL a = sqrt(k), b = k/a;
        for(int i=a; i>; i--)
        {
            LL a0 = k/i, an = k/(i-);
            if(a0 > n) break;       // 如果下限大于n 结束循环
            if(an > n) an = n;      // 如果上限大于n 缩小区间
            res += (k % (a0+) + k % an) * (an - a0) / ;    //求和公式
        }
        for(int i=; i <= n && i <= b; i++)
            res += k%i;
        cout << res <<endl;

    }

    return ;
}