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

算循环

题目大意

求$n$*$m$的矩阵的所有子矩阵面积之和。

数据范围

题解

首先考虑矩阵的横向长度，如果为i，那么有(m-i+1)的选择位置，则横向长度的总贡献为$\sum_{i=1}^{m}$i*($m$-$i$+$1$)，同理，纵向长度的总贡献为$\sum_{i=1}^{n}$i*($n$-$i$+$1$)，则答案为[$\sum_{i=1}^{m}$i*($m$-$i$+$1$)] * [$\sum_{i=1}^{n}$i*($n$-$i$+$1$)]。
那现在时间复杂度为$n$+$m$的，显然会超时。
设$F_n$=$\sum_{i=1}^{n}$i*($n$-$i$+$1$)，则答案即为$F_n$ * $F_m$。
易得$F_n$=$F_{n-1}$+$n^2$+$n$，所以$F_n$=$\sum_{i=1}^{n}$($i^2$+$i$)=$\sum_{i=1}^{n}$$i^2$+$\sum_{i=1}^{n}$$i$

=$n*(n+1)*(2*n+1) \over 6$+$n*(n+1)\over 2$

这样就可以O(1)求出答案了。（除法用逆元）

Code(Pascal)

const
    mo=1000000007;
var
    i,l:longint;
    ans,n,k,m,n2,n6:int64;
function ksm(o,k:int64):int64;
    begin
        ksm:=1;
        while k>0 do
        begin
            if k mod 2=1 then ksm:=ksm*o mod mo;
            o:=o*o mod mo;
            k:=k div 2;
        end;
    end;
begin
    readln(n,m);
    ans:=0;
    k:=0;
    n:=n mod mo;
    m:=m mod mo;
    n2:=ksm(2,mo-2);
    n6:=ksm(6,mo-2);
    k:=(n*(n+1) mod mo)*n2 mod mo;
    k:=k+((n*(n+1) mod mo)*(2*n+1) mod mo)*n6 mod mo;
    k:=k*n2 mod mo;
    ans:=(m*(m+1) mod mo)*n2 mod mo;
    ans:=ans+((m*(m+1) mod mo)*(2*m+1) mod mo)*n6 mod mo;
    ans:=ans*n2 mod mo;
    ans:=ans*k mod mo;
    writeln(ans);
end.