昨天做PKU的排列问题,感觉代码效率应该很高,但就是Time Limit Exceeded,逐步地去改进自己的代码,在本机上对同一组测试数据测试,运行速度明显提高了,但无论怎么提交,就是Time Limit Exceeded,相当相当郁闷,最后把输入输出改成了标准C的stdio.h,结果通过了,上网查相关解释,原来标准C的stdio.h要比标准C++的iostream快一倍,真是长见识了。贴出我的代码,纪念一下自己!感觉这个题目的运行时间全部都让输入输出给占去了。
#include <stdio.h>
#include<algorithm>
using namespace std;
int main()
{
int a[1024];
int m,n,k,ii=0;
int i,j;
scanf("%d",&m);
while(m-->0)
{
scanf("%d%d",&n,&k);
for(j=0; j<n;j++)
{
scanf("%d",&a[j]);
}
if(n==1) { printf("1\n");continue;}
if(n==3) k=k%6;
if(n==2) k=k%2;
while(k>0) //计算下k个排列数
{
for(i=n-1;i>0;i--)
if(a[i-1]<a[i]) break;
if(i>0)
{
int pos=i;
for(j=i+1;j<n;j++)
{ if(a[j]>a[i-1])
{
if(a[j]<a[pos]) pos=j;
}
else break;
}
swap(a[i-1],a[pos]);
}
sort(a+i,a+n);
if(n-i>=4)
{
if(k>=24) { k=k-24; swap(a[n-4],a[n-1]);swap(a[n-3],a[n-2]);}
else if(k>=6){ k=k-6;swap(a[n-1],a[n-3]);}
else if(k>=2){ k=k-2;swap(a[n-1],a[n-2]);}
else k--;
}
else if(n-i==3)
{
if(k>=6){k=k-6;swap(a[n-1],a[n-3]);}
else if(k>=2) {k=k-2;swap(a[n-1],a[n-2]);}
else k--;
}
else if(n-i==2)
{
if(k>=2) {k=k-2;swap(a[n-1],a[n-2]);}
else k--;
}
else k--;
}
for(j=0;j<n;j++)
{
printf("%d ",a[j]);
}
printf("\n");
}
return 0;
}
以后在PKU上做题我就改用stdio.h了。