首页 > 代码库 > C++ P1188 火星人(NOIP2004普及组第3题)

C++ P1188 火星人(NOIP2004普及组第3题)

#include<iostream>
#include<cstdio>
using namespace std;

int N[10005],n,m,time,ap[10005],sign[10005];
//ap用来表示某一元素在排列中出现的次数;
bool get,out;//表示是否已经找到标准排列,是否已经输出;
void DFS(int x)
{
  if(x>n)
  {
    get=1;//找到标准排列;
    time++;
    if(time==m+1)
    {
      for(int i=1;i<=n;i++) printf("%d ",N[i]);
      out=1;
    }
  return;
  }
  for(int i=1;i<=n;i++)
  {
    if(get&&ap[i]<sign[i])//找到了标准排列(注意判断找到和没找到两者的位置很重要) ;
    {
      if(ap[i]<sign[i])
      {
        N[x]=i;//找到后开始生成排列;
        ap[i]++;
        DFS(x+1);
        ap[i]--;
      }
      if(out==1) return;
    }
    if(!get&&i==N[x]&&ap[i]<sign[i])//没找到标准排列,剪枝,只有当每一位上数字相同才向下进行;
    {
      ap[i]++;
      DFS(x+1);
      ap[i]--;
      if(out==1) return;
    }
  }
}

int main()
{
  scanf("%d%d",&n,&m);
  for(int i=1;i<=n;i++)
  {
    scanf("%d",&N[i]);//输入要查找的标准排列;
    sign[i]=1;//每一个数字的出现次数;
  }
  DFS(1);//回溯,查找排列;
  return 0;
}

C++ P1188 火星人(NOIP2004普及组第3题)