类欧几里得

【xsy2126】超级绵羊异或 类欧几里德算法

题意

分析

  根据位运算的位独立性，我们逐位考虑当前位的异或值，即分析每一位的个数是有奇数个，还是偶数个。
  我们从最后一位开始进行分析。答案即为

$$\sum_{i=0}^{n-1}((bi+a)\mod 2)\mod 2=(\sum_{i=0}^{n-1}bi+a)\mod 2$$
特殊的，对于第 $i$ 位，答案为 $$\sum_{i=0}^{n-1}(\lfloor \frac{bi+a}{2^i}\rfloor\mod 2)\mod 2=(\sum_{i=0}^{n-1}\lfloor \frac{bi+a}{2^i}\rfloor)\mod 2$$

  发现我们要求 $O(\log n)$ 个形如

$$calc(n,a,b,c)=\sum_{i=0}^{n}\lfloor\frac{ai+b}{c}\rfloor$$ 的式子的答案。

  当 $a\ge c$ 或者 $b\ge c$ 时，我们显然可以将原式化简。
  设 $a=pc+a',b=qc+b'$ ，则有

$$\begin{aligned}calc(n,a,b,c)&=\sum_{i=0}^{n}\lfloor\frac{ai+b}{c}\rfloor\\&=\sum_{i=0}^{n}\lfloor pi+q+\frac{a'i+b'}{c}\rfloor\\&=p\frac{n(n+1)}{2}+q(n+1)+calc(n,a',b',c)\end{aligned}$$

  当 $a<c$ 且 $b<c$ 时，记 $m=\lfloor{an+b\over c}\rfloor$ ，则有

$$\begin{aligned}calc(n,a,b,c)&=\sum_{i=0}^n\lfloor\frac{ai+b}{c}\rfloor\\&=\sum_{i=0}^n\sum_{j=0}^{m-1}[j< \lfloor\frac{ai+b}{c}\rfloor]\\&=\sum_{i=0}^n\sum_{j=0}^{m-1}[j+1\le \lfloor\frac{ai+b}{c}\rfloor]\\&=\sum_{i=0}^n\sum_{j=0}^{m-1}[cj+c-b\le ai]\\&=\sum_{i=0}^n\sum_{j=0}^{m-1}[cj+c-b-1<ai]\\&=\sum_{i=0}^n\sum_{j=0}^{m-1}[i>\lfloor\frac{ cj+c-b-1}{a}\rfloor]\\&=\sum_{j=0}^{m-1}(n+1-\sum_{i=0}^n[i\le \lfloor\frac{cj+c-b-1}{a}\rfloor])\\&=n\times m-\sum_{j=0}^{m-1}\lfloor\frac{cj+c-b-1}{a}\rfloor\end{aligned}$$

  边界条件为 $a=0$ 。此时

$$calc(n,a,b,c)=\sum_{i=0}^n\lfloor \frac{b}{c}\rfloor=(n+1)\lfloor\frac{b}{c}\rfloor$$

代码

#include <iostream>
using namespace std;

#define LL long long

const int B=61;

LL nT;
LL n,a,b;

inline LL odd(LL x) {
    if (x<0) x=-x;
    return x&1;
}
inline LL binom(LL n) {
    return odd(n*(n+1)/2);
}
LL calc(LL n,LL a,LL b,LL c) {
    if (!a) {
        LL sum=odd(n+1)*odd(b/c);
        return odd(sum);
    }
    else if (a>=c||b>=c) {
        LL _a=a%c,p=(a-_a)/c;
        LL _b=b%c,q=(b-_b)/c;
        LL sum=odd(p)*binom(n)+odd(q*(n+1))+calc(n,_a,_b,c);
        return odd(sum);
    }
    else {
        LL m=(a*n+b)/c;
        LL sum=odd((n)*m)-calc(m-1,c,c-b-1,a);
        return odd(sum);
    }
}

int main(void) {
    #ifndef ONLINE_JUDGE
        freopen("sdchr.in","r",stdin);
//      freopen("sdchr.out","w",stdout);
    #endif

    for (cin>>nT;nT>0;nT--) {
        cin>>n>>a>>b;
        LL sum=0;
        for (int i=0;i<=B;i++)
            sum|=(calc(n-1,b,a,1ll<<i)<<i);
        cout<<sum<<endl;
    }

    return 0;
}