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