牛牛的作业薄上有一个长度为 n 的排列 A,这个排列包含了从1到n的n个数,但是因为一些原因,其中有一些位置(不超过 10 个)看不清了,但是牛牛记得这个数列顺序对的数量是 k,顺序对是指满足 i < j 且 A[i] < A[j] 的对数,请帮助牛牛计算出,符合这个要求的合法排列的数目。
输入描述:
每个输入包含一个测试用例。每个测试用例的第一行包含两个整数 n 和 k(1 <= n <= 100, 0 <= k <= 1000000000),接下来的 1 行,包含 n 个数字表示排列 A,其中等于0的项表示看不清的位置(不超过 10 个)。
输出描述:
输出一行表示合法的排列数目。
输入例子:
5 5
4 0 0 2 0
输出例子:
2
#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <stdlib.h>
#include <limits.h>
#include <math.h>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn = 100 + 3;
int a[maxn]; //存放的原始数据
int b[12]; //存放未知的位置数据
int c[maxn]; //b每个排列放到对应的位置的顺序对
int n,k;
//自己实现的全排列
void perm(int a[],int l,int r){
if(l==r){
for(int i=0;i<=r;i++){
printf("%d ",a[i]);
}
printf("\n");
}else{
for(int i=l;i<=r;i++){
swap(a[i],a[l]);
perm(a,l+1,r);
swap(a[i],a[l]);
}
}
}
int FindNum(int s[],int len){
int num=0;
for(int i=0;i<len-1;i++){
for(int j=i+1;j<len;j++){
if(s[i]<s[j]){
num++;
}
}
}
return num;
}
int main(){
scanf("%d %d",&n,&k);
int len=0;
int vis[maxn];
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
vis[a[i]]=1;
c[i]=a[i];
}
for(int i=1;i<=n;i++){
if(!vis[i]) b[len++]=i;
}
int ans=0;
do {
for(int i=0,j=0;i<n;i++){
if(a[i]==0){
c[i]=b[j++];
}
}
if(FindNum(c,n)==k) ans++;
}while(next_permutation(b,b+len));
printf("%d\n",ans);
return 0;
}
但是这样的时间复杂度是10!*100*50,应该会超时的,不过居然AC了。
然后发现可以通过预处理和分段计算的方法,把时间复杂度降低为10!*55;
#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <stdlib.h>
#include <limits.h>
#include <math.h>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn = 100 + 3;
int a[maxn]; //存放的原始数据
int b[12]; //存放未知的位置数据
int c[maxn]; //b每个排列放到对应的位置的顺序对
int d[maxn]; //存放已知的位置数据
int n,k;
int below[12][maxn];
int up[12][maxn];
int FindNum(int s[],int len){
int num=0;
for(int i=0;i<len-1;i++){
for(int j=i+1;j<len;j++){
if(s[i]<s[j]){
num++;
}
}
}
return num;
}
int main(){
scanf("%d %d",&n,&k);
int len1=0;
int len2=0;
int vis[maxn];
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
vis[a[i]]=1;
c[i]=a[i];
if(a[i]!=0) d[len2++]=a[i];
}
for(int i=1;i<=n;i++){
if(!vis[i]) b[len1++]=i;
}
//预处理,找出比B[i]在第i个位置时,左边比它小的个数,右边比它大的个数
for(int i=0;i<len1;i++){
int num=0,t=0;
for(int j=0;j<n;j++){
if(a[j]!=0){
if(a[j]<b[i]) { ++num;}
}else{
below[t++][b[i]]=num;
}
}
num=0;
t=len1-1;
for(int j=n-1;j>=0;j--){
if(a[j]!=0){
if(a[j]>b[i]) { ++num;}
}else{
up[t--][b[i]]=num;
}
}
}
int ans=0;
int tmp=FindNum(d,len2);
//结果为未知的序列对个数+已知的位置序列对个数+未知的与已知的序列对个数
do {
int sum=tmp;
sum+=FindNum(b,len1);
for(int i=0;i<len1;i++){
sum+=below[i][b[i]]+up[i][b[i]];
}
if(sum==k) ans++;
}while(next_permutation(b,b+len1));
printf("%d\n",ans);
return 0;
}