codeforces 821 B Okabe and Banana Trees

题目：http://codeforces.com/contest/821/problem/B

大致题意：

给两个正整数m,b,画出一条直线y = -x/m + b，求直线下的一个矩形使得获取的香蕉数最多，每个点(x,y)的香蕉数目为 x+y。

解法：

可推得公式，矩形右上角为(x,y)的香蕉总数为

(x+1) * x / 2 * (y+1) + (y+1) * y /2 * (x+1)

然后，依次遍历直线上的格点，求出矩形右上角为格点坐标的香蕉总数，取最大值。

代码：

#include <iostream>
#include <algorithm>
using namespace std;
int main(){
long long m,b;
cin >> m >> b;
long long ans = -1;
for(long long i = 0 ; i <= b; i++){
long long j = m * (b-i);
ans = max(ans,
(1+i) * i / 2 * (j+1) + (1+j) * j /2 * (i+1));
}
cout << ans << endl;
}