深夜切题——等差对

10674 等差对

时间限制:1000MS  内存限制:65535K

Description

   今天是一个特别的日子，百年一遇的光棍节，2011.11.11，xym收到一个装着礼物的信封，是一位mm的XX书，里面是两个棒棒糖和一封信。
信里是一道智力题：
    定义如果<x0,y0>和<x1,y1>满足x0-x1=y0-y1，则称这两个为等差对。
mm的问题是，问在<x,y>（0<=x,y<=n）<0,0>,<0,1>…<1,0>,<1,1>…<2,0>,
<2,1>…<n,0>…<n,1>…这(n+1)^2个有序对中存在多少个等差对？
    但是xym因为昨晚听他们班的女生唱歌太晚睡觉了，严重影响状态，现在他只能请求各位scau未来的希望帮助他解决这个问题。

输入格式

第一行输入一个整数T(T<=1000)，表示case数。
下面T行有T个case，每个case只有一个整数n（1<=n<=10^9），表示0<=x,y<=n;但是由于测试的时候发现用scanf("%lld",&n)有bug，所以为了修正这个bug，请各位用long long读入的同学在读入前先赋0给变量，
例如 long long n=0; scanf("%lld",&n);

输出格式

每个case输出一行,表示等差对的数量，这个结果可能很大，只需最后结果%20111111即可。

输入样例

2
2
100

输出样例

5
338350

提示

注意不要让数据溢出，及时取模的处理

来源

 by 201030720425 

C2 (n+1) + 2*C2 n + ... + 2*C2 2 = ... = (2*n+1)*(n+1)*n/6

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
const int mod=20111111;
void put(long long int n)
{
    long long int sum=0;
    if((n+1)*n%6==0)sum=(((n+1)*n/6)%mod)*((2*n+1)%mod);
    else sum=(((2*n+1)/3)%mod)*((n*(n+1)/2)%mod);
    printf("%lld\n",sum%mod);
}
//#define local
int main()
{
#ifdef local
    freopen("input.txt","r",stdin);
#endif
    int T;
    long long int n;
    scanf("%d",&T);
    while(T--){
        n=0;
        scanf("%lld",&n);
        put(n);
    }
    return 0;
}