Luogu P1497题解

题目描述：
有一个n * n的方形区域，同时你有k个木牛流马，同时这k个木牛流马有h种颜色，现在规定两个木牛流马不能同时存在于一行或者一列里面，现在问有几种放置方法？
题解
组合数学问题，这个题目整体的思想还是分步骤进行
1.先认为是一种颜色，找出有几种选择位置的方法。
2.再对这些木牛流马涂色。
如果我们先选择列，那就是有C(n,k)种方法，如果选择列的基础上再去选择就有A(n,k)种方法,然后我们再去分配这些颜色，那就是每次都有C(n,i)种选择，但是注意每次都要减去i的数目。
AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
using namespace std;
ll calculate_a(int n,int k)
{
    ll x = 1;
    for(int i = 1;i <= k;i++) x *= (n + 1 - i);
    return x;
}
ll calculate_c(int n,int k)
{
    ll x = 1;
    for(int i = 1;i <= k;i++) x *= (n + 1 - i);
    for(int i = 1;i <= k;i++) x /= i;
    return x;
}
int main()
{
    int n,k,h;
    ll ans_a,ans_c,ans = 1;
    cin >> n >> k >> h;
    ans_a = calculate_a(n,k);
    ans_c = calculate_c(n,k);
    ans = ans_a * ans_c;
    // cout << ans_a << ' ' << ans_c << endl;
    // cout << ans << endl;
    for(int i = 1;i <= h;i++){
        int color;
        cin >> color;
        ans *= calculate_c(k,color);
        k -= color;
    }
    cout << ans << endl;
    return 0;
}