uva 11538 Chess Queen

点击打开链接

题意：给定一个n*m的矩阵，问有多少种方法放置两个相互攻击的皇后？规定在同一行同一列和同对角线的能够相互攻击

思路：

1 先考虑同一行的情况，n行就有n种情况，每一行有m*（m-1）种，总的是n*m*(m-1);

2 考虑同一列的情况，m列就是m种情况，每一列有n*(n-1)种，总的是m*n*(n-1);

3 最后考虑同对角线的情况，因为n > m和 n <= m的情况是不一样的，我们先假设 n <= m

   通过画图我们知道，所有的/对角线，从左到右的长度分别为1 2 3 ... n-1
n n ...n n(有m-n+1个) n-1 ... 1

   那么总的就是1*(1-1)+2*(2-1)+...(n-1)*(n-1-1)+(m-n-1)*n*(n-1)+(n-1)*(n-1-1)+...+2*(2-1)+1*(1-1)

   因为还有\的情况，因为总的还要乘2

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

typedef long long int64;

int64 n , m;

int64 solve(){
int64 ans = 0;
if(n > m)
swap(n , m);
// same row
ans += n*m*(m-1);
// same col
ans += m*n*(n-1);
// same duijiao
ans += 2*(2*((n-1)*n*(2*n-1)/6-n*(n-1)/2)+(m-n+1)*n*(n-1));
return ans;
}

int main(){
while(cin>>n>>m && n+m){
cout<<solve()<<endl;
}
return 0;
}