51nod 3034 矩形的数量 V2

题目

一个m乘n的方格图，有多少种形状不同的矩形。1x2和2x1算同一种。

例如：4*3的方格中共有9种。

1x1,1x2,1x3,2x2,2x3,3x3,4x1,4x2,4x3

输入
共一行：2个数m,n，对应方格图的长宽（1 <= m, n <= 1e9）
输出
输出对应的矩形种类 Mod 1e9+7。
数据范围
1 <= m, n <= 1e9
输入样例
4 3
输出样例
9
样例解释
9种分别是：1x1,1x2,1x3,2x2,2x3,3x3,4x1,4x2,4x3

解题思路

用两个循环加出最后结果再mod1e9+7

代码

#include <bits/stdc++.h>
#include<iostream>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>
#include <list>
#include <utility>
#include<cstring>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <bitset>
#include <iterator>
#define INT_MAX 0x7fffffff
#define INT_MIN 0x80000000
const int MOD = 1E9+7;
const int N = 100000+5;
using namespace std;


int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    //freopen("input.txt","r",stdin);
    // IO
    long long int n,m;
    cin >> n >> m;
    if(n < m){
        swap(n,m);
    }
    long long int cnt = 0;
    for(int i = 0; i <m; i++ ){
        for(int j = i; j <m; j++ ){
            //cout << i << ' ' << j << endl;
            cnt++;
        }
    }
    for(int i = m; i < n;i++){
        for(int j = 0; j <m; j++){
            //cout << i << ' ' << j << endl;
            cnt++;
        }
    }
    cout << cnt%MOD;

    return 0;
}