AcWing - 5287. 数量 - 组合数求解+思维

最多只有k个位置满足，ai != i
所以从0-k进行遍历，对于kj：
0时，所有位置上的数字都等于i，只有一种
1时，没有
2时，排列只有两种，1 2 或者 2 1，只有一种情况满足
同理3和4可以分别只有2，9种
具体细节见代码

#include <math.h>
#include <stdio.h>
#include <algorithm>
#include <cstring>
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
#define de(x) cout << x << " ";
#define sf(x) scanf("%d", &x);
#define Pu puts("");
#define ll long long
// int n, m, ans;
ll n, k, ans;
ll C(ll x, ll y) {  // 组合数量
    ll res = 1;
    for (ll i = 1; i <= y; i++) {
        res *= (x - i + 1);
        res /= i;
    }
    return res;
}
int main() {
    cin >> n >> k;
    ans = 1;
    // 最多只有k个位置满足，ai != i
    // 所以从0-k进行遍历
    for (ll i = 2; i <= k; i++) {
        if (i == 2) {
            ans += C(n, i) * 1;
        } else if (i == 3) {
            ans += C(n, i) * 2;
        } else if (i == 4) {
            ans += C(n, i) * 9;
        }
    }
    cout << ans << endl;
    return 0;
}