[BZOJ3817] Sum

Description

给出n,r
求

$$\Large \sum\limits_{i=1}^{n}(-1)^{\lfloor i\sqrt r\rfloor}$$
T组数据

T,r<=10000,n<=10^9

Solution

转化式子
$(-1)^a=1-2\times (a\mod 2)$
$a\mod 2 =a-2\times \lfloor a/2\rfloor$

原式化为$n-2\sum\limits_{i=1}^{n}{\lfloor i\sqrt r\rfloor}+4\sum\limits_{i=1}^{n}{\lfloor {i\sqrt r\over 2}\rfloor}$

如果r是完全平方数就可以直接计算

考虑如何计算$\sum\limits_{i=1}^{n}{\lfloor i\sqrt r\rfloor}$
即求一个斜率为无理数的正比例函数下整点个数

令$t=\sqrt r$
将${\lfloor i\sqrt r\rfloor}$表示成${\lfloor i\times {at+b\over c}\rfloor}$的形式,a,b,c都是整数，一开始a=1,b=0,c=1

现在考虑如何计算$\sum\limits_{i=1}^{n}{\lfloor i\times {at+b\over c}\rfloor}$

当${at+b\over c}\geq 1$
${\lfloor i\times {at+b\over c}\rfloor}={\lfloor i\times ({at+b\over c}-\lfloor{at+b\over c}\rfloor)\rfloor+i\times\lfloor{at+b\over c}\rfloor}$

后面这部分可以直接计算出，前面就更新b，继续递归处理

当$0<{at+b\over c}<1$
既然要求的是一个原点为顶点的直角三角形中整点个数，且斜率小于1，邻边长为n

我们可以将x,y轴互换（关于直线y=x对称）

那么就变成了求上方白色三角形中整点个数，它与红色三角形中整点个数是一样的
这时对边变成了n，而邻边变成了${\lfloor n\times {at+b\over c}\rfloor}$
把这个看成是新的n

同时斜率变为原来的倒数，即${c\over at+b}$
分母有理化
${c\over at+b}={act-bc\over a^2r-b^2}$
又得出了新的a,b,c递归处理

注意有可能会爆Long long 分式需要上下同除gcd(a,b,c)

似乎这个过程很像类欧几里得算法

复杂度分析:
设$k={at+b\over c}$
可以发现上面的过程是轮流交替的

如果${1\over 2}<k<1$,那么$1<{1\over k}<2$，${k\over {1\over {1\over k}-1}}=1-k<{1\over 2}$
那么k大于1/2时减小速度至少是1/2的
如果$k<{1\over 2}$，那么n的减小速度又至少是1/2了

因此总的复杂度是$O(\log N)$的
好像r如果是完全平方数就会算重来着。。。因此要放在之前特殊计算

Code

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
#define LL long long
using namespace std;
LL n,r;
double q;
LL gcd(LL x,LL y)
{
    return(!y)?x:gcd(y,x%y);
}
LL calc(LL a,LL b,LL c,LL n)
{
    if(n==0) return 0;
    if(n==1) return (LL)(a*q+b)/c;
    LL v=gcd(gcd(a,b),c);
    a/=v,b/=v,c/=v;
    LL p=(a*q+b)/c,s=0;
    if(p==0)
    {
        LL n1=((a*q+b)/c*n);
        return (n1*n-calc(a*c,-b*c,a*a*r-b*b,(LL)n1));
    }
    else
    {
        return p*(n*(LL)(n+1)/2)+calc(a,b-c*p,c,n);
    }
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        scanf("%lld%lld",&n,&r);
        q=sqrt(r);
        int t=(int)q;
        if(t*t==r) 
        {
            if(t%2==0) printf("%lld\n",n);
            else printf("%lld\n",n-2*((n+1)/2));
        } 
        else
        {
            printf("%lld\n",n-(LL)2*calc(1,0,1,n)+(LL)4*calc(1,0,2,n));
        }
    }
}