「组合计数」CodeForces - 888D

最新推荐文章于 2021-08-26 21:48:39 发布

Orion233

最新推荐文章于 2021-08-26 21:48:39 发布

阅读量372

点赞数

分类专栏：代码强迫症

本文链接：https://blog.csdn.net/orion_pistachio/article/details/78534130

版权

代码强迫症专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题意&分析：
对于一个给定数列，求满足 ai = i 的元素个数不少于 n-k 个的数列的个数。换句话说，就是求最多有 k 个元素不满足 ai = i 的数列有几个。
分析一下用排列数得出一下结论：
k = 1 时，ans = 1；
k = 2 时， ans = 1 + C(2,n);
k = 3 时 , ans = 1 + C(2,n) + C(3,n)*2 (其中三个元素排列不在自己位置上的情况只有两种，即2 3 1 和 3 1 2)；
k = 4 时， ans = 1 + C(2,n) + C(3,n) * 2 + C(4,n) * 9 (9同上理)；

代码如下：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <cctype>
#include <fstream>
#define INF 0x3f3f3f3f
#define EPS 1e-6
#define TEST cout<<"stop here"<<endl
using namespace std;
typedef long long ll;
const ll mod = 1e9 + 7;

ll n,k;
int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);

    while(cin>>n>>k){
        ll ans = 0;
        if(k == 1){
            ans = 1;
        }
        else if(k == 2){
            ans = 1 + n*(n-1)/2;
        }
        else if( k == 3){
            ans = 1 + n*(n-1)/2 + 2*n*(n-1)*(n-2)/6;
        }
        else if(k == 4){
            ans = 1 + n*(n-1)/2 + 2*n*(n-1)*(n-2)/6 + 9*n*(n-1)*(n-2)*(n-3)/24;
        }
        cout<< ans <<endl;
    }

    return 0;
}