[编程题] 数列还原

牛牛的作业薄上有一个长度为 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;
}