C. Dreamoon and Sums
题意:输入a,b,根据a和b计算所有nice integer的和。nice integer是这样定义的。。如果mod(x,b)不为0,div(x,b)/mod(x,b)=k,k属于[1,a],那么符合上述条件的x为nice integer。
思路:推公式。我们可以这样推出一个矩阵来,所有元素都是一个nice integer,把每个元素相加即可。首先写下一行i(i=1~b-1)作为mod(x,b)的结果(不允许为0)。然后往下写数,每次在i下面写下i+i*j*b(j=1~a),然后这个a行b-1列的矩阵就是所有的nice integer了。怎么求和呢,很容易发现这个矩阵无论是行还是列都是等差数列,直接套等差数列求和公式即可。
不得不说的是,这题比赛时我跪了,原因在取模。。必须注意到先除以2和先取模的结果不一定相同,一定要处理好。我的处理办法是先化简公式,然后把除以2放在取模前面。
#include <iostream>
#include <stdio.h>
#include <cmath>
#include <algorithm>
#include <iomanip>
#include <cstdlib>
#include <string>
#include <memory.h>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <ctype.h>
#include <sstream>
#define INF 1000000000
#define ll long long
#define min3(a,b,c) min(a,min(b,c))
#define max3(a,b,c) max(a,max(b,c))
#define MAXN 100010
using namespace std;
int main(){
ll a,b;
while(cin>>a>>b){
ll tmp=b*(b-1)/2;
tmp%=1000000007;
ll tmp2=a*(a+1)/2;
tmp2%=1000000007;
tmp2*=b;
tmp2+=a;
tmp2%=1000000007;
ll ans=tmp*tmp2;
ans%=1000000007;
cout<<ans<<endl;
}
return 0;
}