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;
}