题目
样例输入:
167 198
样例输出:
906462341
数据范围:
剖解题目
给一个矩阵,求矩阵所有子矩阵的面积和,取模。
思路
第一眼,20分肯定是原来暴力。
第二眼,40分的不就是最后那两重循环去掉嘛。。。
然后脑补。
想到了一个递推式,但是这个递推是是二维的并且还有加数是两个数的乘积。
发现乘积可以代替。
于是很自然的想到了矩阵乘法+快速幂大法。
然而二维的怎么打啊。。。。。
就炸了。
解法
40分不说,这道题确实是可以用矩阵乘法快速幂弄出来,只不过我不会。。。。
60分:
如果固定了矩形的左上角(也就是 i 和 j 的枚举),那么剩下的就可以看成是一些共用一个角落的矩形的面积和。 设最大的矩形是 N ×M 的,那么
设
S(n,m)=∑x=1n∑y=1mxy=n(n+1)2∗m(m+1)2
答案为
∑i=1n∑j=1mS(n−i+1,m−j+1)
逆元提前处理,时间 O(n2)
80%:
首先答案还同样可以等于
∑ni=1∑mj=1S(i,j)
。展开后得到
14∑ni=1i(i+1)∑mj=1j(j+1)
。
计算每个
∑
部分。时间
O(n)
100%:
设
f(n)=∑ni=1i2∗∑ni=1i
.
利用平方和公式以及等差数列求和公式得到
f(n)=16n(2∗n+1)(n+1)+12n(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);
}