codeforces888D Almost Identity Permutations 排列组合

题目链接：戳这里

题目大意：给出1个数n，求1~n的排列中满足最多有 k 个元素不满足 ai = i 的数列有几个。

题解：考虑到k只有4，所以可以直接分类讨论，这个很好手推...

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
LL read()
{
    char c;LL sum=0,f=1;c=getchar();
    while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}
    return sum*f;
}
LL n,k;
int main()
{
    n=read();k=read();
    switch (k)
    {
        case 1:puts("1");break;
        case 2:printf("%lld\n",n*(n-1)/2+1);break;
        case 3:printf("%lld\n",n*(n-1)*(n-2)/3+n*(n-1)/2+1);break;
        case 4:printf("%lld\n",3*n*(n-1)*(n-2)*(n-3)/8+n*(n-1)*(n-2)/3+n*(n-1)/2+1);break;
    }
    return 0;
}