Chess Queen【数学】

Chess Queen

 UVA - 11538 

题目传送门

题目大意：输入两个整数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;
}