矩形的数量 V3 51NOd-3036

题目链接：矩形的数量V3

一个m乘n的方格图，有多少个大小不同的正方形。
例如：4*3的方格中共有20个，其中：
1x1 12个
2x2 6个
3x3 2个

输入
共一行：2个数m,n，对应方格图的长宽（1 <= m, n <= 1e9）
输出
输出对应的矩形种类 Mod 1e9+7。
数据范围
1 <= m, n <= 1e9

输入样例
4 3
输出样例
20
样例解释
1x1 12个
2x2 6个
3x3 2个

最初的思路：
假设长为p，宽为t
矩形中的大小不同的正方形数量经过观察可看出是Σ(p-i)(t-i),如果从i遍历到t的话很遗憾会超时，1e9，最后一组数据死活过不去。

#include <iostream>
#include <cmath>
#include <algorithm> 
using namespace std;
const int mod=1000000007;
long long m,n;
long long cnt=0;
int main(){
	scanf("%lld%lld",&m,&n);
	long long p=max(m,n);
	long long t=min(m,n);
//	cnt+=(m*n)%mod;
	for(long long i=0;i<t;i++){
		//long long z=p-i+1,h=t-i+1;
		cnt=(cnt+((p-i)*(t-i))%mod)%mod;
	}
	cout<<cnt%mod<<endl;
	return 0;
} 

但是观察累加公式可知，可以将这个公式变一变形，变成等差或者什么公式就不需要从头到尾遍历了，只需要一个公式既可以求出最终结果。
令q=p-t
Σ(p-i)(t-i) == Σ(p-t+t-i)(t-i) == Σ(t-i)(t-i)+(p-t)(t-i)------------i=0,1,2,3…t
可知对每次的长宽, p-t是一个常数
到这一步，还有两个关键公式要清楚
①平方和公式 ----------------Σn²=n(n+1)×(2×n+1)/6----------n=1,2,3…n
②等差数列求和公式 --------Σn=(1+n)×n/2-------------------n=1,2,3…n
这样就把关于i的循环去掉了，直接求出最终结果：
ans=(t×(t+1)×(2×t+1)/6+(p-t) ×(1+t)×t/2)%1000000007
好吧我承认，这道题真正难的地方其实是取模，相当麻烦容易出错

#include <bits/stdc++.h>
using namespace std;

const long long N=1e9+7;
long long p,q;
long long ans;
int main(){
	cin>>p>>q;
	long long m=max(p,q);
	long long n=min(p,q);
	long long s=(((m-n)%N)*((1+n)*n/2)%N)%N;
	long long f1=(n%(N*6)*(n+1)%(N*6))%(6*N);
	long long f2=(f1*((2*n+1)%(6*N)))%(6*N);
	ans=(f2/6+s)%N;
	cout<<ans<<endl;
	return 0;
} 

取模的顺序是要注意的，一定要优先对结果偏大的部分先取模，如果对小的先取模，小的再去乘一个大的可能会溢出，再取模就是错的。所以要先对大的取模使其变小。
当然无所谓顺序的话就要活用括号，让大的优先取模在进行运算。