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);elset2+=k*(k-1)*(q-k-1);ll ans=t1+t2*2;cout<<ans<<endl;}return 0;
}

?