版权声明:版权声明:本文为博主原创文章,转载请注明出处。 https://blog.csdn.net/YT201758501112/article/details/83898869
Chess Queen
题目大意:输入两个整数n,m,在n行m列的棋盘中放入白黑两个棋子,棋子在同一行、同一列或同一对角线上能相互进攻,问有多少种摆放方案。
AC代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>
#include <set>
#include <utility>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define rep(i,l,r) for(int i=l;i<=r;i++)
#define lep(i,l,r) for(int i=l;i>=r;i--)
#define ms(arr) memset(arr,0,sizeof(arr))
//priority_queue<int,vector<int> ,greater<int> >q;
const int maxn = (int)1e5 + 5;
const ll mod = 1e9+7;
int main()
{
//freopen("in.txt", "r", stdin);
//freopen("out.txt", "w", stdout);
ios::sync_with_stdio(0),cin.tie(0);
ll n,m;
while(cin>>n>>m&&n+m)
{
ll t1=(n+m-2)*(n*m); //在行和列上的情况
ll t2=0,k=min(n,m),q=max(n,m);
for(ll i=2;i<=k;i++)
t2+=i*(i-1); //一个方向上对角线的情况
t2=t2*2;
if(n==m)
t2-=n*(n-1);
else
t2+=k*(k-1)*(q-k-1);
ll ans=t1+t2*2;
cout<<ans<<endl;
}
return 0;
}