题目链接:戳这里
题目大意:给出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;
}