P1548 [NOIP1997 普及组] 棋盘问题（洛谷）

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m,cnt1=0,cnt2=0;
    cin>>n>>m;//输入列长(n) 和 输入行长(m)
    for(int i=1;i<=n;i++){//记录的正方形 或 长方形的
        for(int j=1;j<=m;j++){//记录的正方形 或 长方形的行长
            for(int k=i;k<=n;k++){//因为从上边开始,所以从第一个记录的正方形 或 长方形的最下一行开始记录更方便
                for(int p=j;p<=m;p++){//因为从左边开始,所以从第一个记录的正方形 或 长方形的最右一行开始记录更方便
                    if(k-i==p-j){//列长等于行长是正方形
                        cnt1++; 
                    }
                    else{//其余的都是长方形
                        cnt2++;
                    }
                }
            }
            
        }
    }
    cout<<cnt1<<' '<<cnt2;//输出正方形的个数 和 长方形的个数
    return 0;
}