【JZOJ4819】【NOIP2016提高A组模拟10.15】算循环

题目描述

输入

输出

样例输入

167 198

样例输出

906462341

数据范围

解法

令$f(n)=\sum_{i=1}^ni$，$g(n)=\sum_{i=1}^ni^2$
易得$ans=\sum_{i=1}^n \sum_{j=1}^m f(n-i+1)*f(m-j+1)$
等价于$ans=\sum_{i=1}^n \sum_{j=1}^m f(i)*f(j)$
显然$f(n)=n*(n-1)/2$；
拆开得$ans= \frac{1}{4}\sum_{i=1}^n \sum_{j=1}^m i*(i+1)*j*(j+1)$
再得

$$ans= \frac{1}{4}\sum_{i=1}^n i*(i+1)*\sum_{j=1}^m j*(j+1) =\frac{1}{4}\sum_{i=1}^n*(f(i)+g(i))*\sum_{j=1}^m*(f(j)+g(j))$$
其中 $g(n)=\frac{1}{6}n(n+1)(2n+1)$

---

时间复杂度为$O(log)$，逆元有复杂度。

代码

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#define ll long long
#define ln(x,y) ll(log(x)/log(y))
#define sqr(x) ((x)*(x))
using namespace std;
const char* fin="loop.in";
const char* fout="loop.out";
const ll inf=0x7fffffff;
const ll mo=1000000007;
ll n,m,i,j,k,l,tmp,tmd,num,ans;
ll qpower(ll a,ll b){
    ll c=1;
    while (b){
        if (b&1) c=a*c%mo;
        a=a*a%mo;
        b>>=1;
    }
    return c;
}
ll N(int a){
    return qpower(a,mo-2);
}
ll sum(ll st,ll num){
    st%=mo;
    num%=mo;
    ll en=(st+num-1)%mo;
    return (st+en)%mo*num%mo*N(2)%mo;
}
ll xsum(ll n){
    n%=mo;
    return n*(n+1)%mo*(2*n+1)%mo*N(6)%mo;
}
ll count(ll v){
    return (sum(1,v)+xsum(v))%mo;
}
int main(){
    freopen(fin,"r",stdin);
    freopen(fout,"w",stdout);
    scanf("%lld%lld",&n,&m);
    ans=count(n)*count(m)%mo*N(4)%mo;
    printf("%lld",ans);
    return 0;
}

启发

$\sum$的运算性质

1.$\sum(a+b)=\sum a+\sum b$
2.$\sum_a \sum_b a*b=\sum_a a*\sum_b b$
3.$\sum_i k*f(i)=k*\sum f(i)$

$\sum_{i=1}^ni^2$公式

$\sum_{i=1}^ni^2=\frac{1}{6}n(n+1)(2n+1)$
证明：
利用数学归纳法检验。
设$g(n)=\sum_{i=1}^ni^2$；
先有

$$g(1)=\frac{1}{6}*1*2*3=1=\sum_{i=1}^ni^2$$
如果 $g(x)$满足 $g(x)=\frac{1}{6}x(x+1)(2x+1)$；
则 $$g(x+1)=\frac{1}{6}x(x+1)(2x+1)+(x+1)^2\\=\frac{1}{6}(x+1)(6x+6+x(2x+1))\\=\frac{1}{6}(x+1)(2x^2+7x+6)\\=\frac{1}{6}(x+1)(x+2)(2x+3)\\=\frac{1}{6}(x+1)[(x+1)+1][2(x+1)+1]$$
综上得证。