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

题目

样例输入：

167 198

样例输出：

906462341

数据范围：

---

剖解题目

给一个矩阵，求矩阵所有子矩阵的面积和，取模。

---

思路

第一眼，20分肯定是原来暴力。
第二眼，40分的不就是最后那两重循环去掉嘛。。。
然后脑补。
想到了一个递推式，但是这个递推是是二维的并且还有加数是两个数的乘积。
发现乘积可以代替。
于是很自然的想到了矩阵乘法+快速幂大法。
然而二维的怎么打啊。。。。。
就炸了。

---

解法

40分不说，这道题确实是可以用矩阵乘法快速幂弄出来，只不过我不会。。。。
60分：
如果固定了矩形的左上角（也就是 i 和 j 的枚举），那么剩下的就可以看成是一些共用一个角落的矩形的面积和。 设最大的矩形是 N ×M 的，那么
设

$$S(n,m)=\sum^n_{x=1}\sum^m_{y=1} xy=\frac{n(n+1)}{2}*\frac{m(m+1)}{2}$$
答案为
$$\sum^n_{i=1}\sum^m_{j=1} S(n-i+1,m-j+1)$$
逆元提前处理，时间 $O(n^2)$

80%：
首先答案还同样可以等于$\sum_{i=1}^n\sum_{j=1}^mS(i,j)$。展开后得到$\frac{1}{4}\sum_{i=1}^ni(i+1)\sum_{j=1}^mj(j+1)$。
计算每个$\sum$部分。时间$O(n)$

100%：
设$f(n)=\sum_{i=1}^ni^2*\sum_{i=1}^ni$.
利用平方和公式以及等差数列求和公式得到

$$f(n)=\frac{1}{6}n(2*n+1)(n+1)+\frac{1}{2}n(n+1)$$
时间复杂度O(1).

---

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define down(i,a,b) for(int i=a;i>=b;i--) 
#define ll long long

using namespace std;

const ll mo=1e9+7;
ll n,m;
ll tot1,tot2,ans;

ll qsm(ll x,ll y)
{
    ll t=1;
    while (y){
        if (y&1) t=t*x%mo;
        x=x*x%mo;
        y>>=1;
    }
    return t;
}
int main()
{
    freopen("loop.in","r",stdin);
    freopen("loop.out","w",stdout);
    scanf("%lld%lld",&n,&m);
    n%=mo; m%=mo;
    tot1=((n*(n+1)%mo*(2*n+1)%mo*qsm(6,mo-2))+((n+1)*n%mo*qsm(2,mo-2)%mo))%mo;
    tot2=((m*(m+1)%mo*(2*m+1)%mo*qsm(6,mo-2))+((m+1)*m%mo*qsm(2,mo-2)%mo))%mo;
    ans=tot1*tot2%mo*qsm(4,mo-2)%mo;
    printf("%lld",ans);
}