F. 实验7_21_编码、解码数组

设有一个整型数组a中存放n个互不相同的整数，a中元素取值范围为0到n-1。a数组的编码定义如下：a[0]的编码为0,放入b[0]；ai的编码为在a[0]、a[1]、…、a[i-1]中比a[i]的值小的数的个数，放入b[i]。例如：a[6]={4，3，0，5，1，2}时，则b[6]={0，0，0，3，1，2}。你有两个任务，任务一是编码，即已知n与数组a，求数组b；任务二是解码，即已知n与数组b，求数组a。
输入与输出要求：首先输入两个整数n和flag。n代表数组元素的个数（1<=n<=100），flag代表任务类型(flag＝1代表编码任务，flag＝2代表解码任务)。然后是n个整数，当flag=1时，这n个数即代表数组a的内容；当flag=2时，这n个数即代表数组b的内容。
输出n个整数，当flag=1时即为编码后数组b的内容，当flag=2时，即为解码后数组a的内容。每个整数用空格分开，最后一个整数后是换行符。
程序运行效果：
Sample 1:
5↙
1↙
2 1 3 0 4↙
0 0 2 0 4
Sample 2:
8↙
2↙
0 1 2 0 1 1 4 5↙
3 6 7 0 2 1 4 5

#include<stdio.h>
int main()
{
	int flag,n;
	scanf("%d%d",&n,&flag);
	int a[n],b[n];
	int cnt=0;
	if(flag==1)
	{
		for(int i=0;i<n;i++)
		{
			scanf("%d",&a[i]);
		}
		for(int i=0;i<n;i++)
		{
			cnt=0;
			for(int j=0;j<i;j++) 
			{
				if(a[j]<a[i])
				{
					cnt++;
				}
			}
			b[i]=cnt;
		}
		for(int i=0;i<n;i++)
		{
			if(i<n-1)
			{
				printf("%d ",b[i]);
			}
			else
			{
				printf("%d\n",b[i]);
			}
		}
	}
	cnt=0;
	if(flag==2)
	{
		for(int i=0;i<n;i++)
		{
			a[i]=-1;
		}
		for(int i=0;i<n;i++)
		{
			scanf("%d",&b[i]);
		}
		//a[n-1]=b[n-1];
		int number[n];
		for(int i=0;i<n;i++)
		{
			number[i]=i;
		}
		//number[b[n-1]]=-1;
		for(int i=n-1;i>=0;i--)
		{
			if(number[b[i]]>=0&&number[b[i]]<=n-1)
			{
				a[i]=number[b[i]];
				for(int k=b[i];k<n-1-cnt;k++)
				{
					number[k]=number[k+1];
				}
				cnt++;
			}
		}
		for(int i=0;i<n;i++)
		{
			if(i<n-1)
			{
				printf("%d ",a[i]);
			}
			else if(i==n-1)
			{
				printf("%d\n",a[i]);
			} 
		}
	}
	return 0;
}

简化版

#include<stdio.h>
int main()
{
	int flag,n;
	scanf("%d%d",&n,&flag);
	int a[n];
	int b[n];
	if(flag==1)
	{
		for(int i=0;i<n;i++)
			scanf("%d",&a[i]);
		for(int i=0;i<n;i++)
		{
			int cnt=0;
			for(int j=0;j<i;j++)
			{
				if(a[i]>a[j])
				{
					cnt++;
				}
			}
			b[i]=cnt;
		}
		for(int i=0;i<n;i++)
		{
			if(i<n-1)
				printf("%d ",b[i]);
			else
				printf("%d\n",b[i]);
		}
	}
	if(flag==2)
	{
		int number[n];
		for(int i=0;i<n;i++) 
			number[i]=i;
		for(int i=0;i<n;i++)
			scanf("%d",&b[i]);
		for(int i=n-1;i>=0;i--)
		{
			a[i]=number[b[i]];
			for(int j=b[i]+1;j<n;j++)
			{
				number[j-1]=number[j];
			}
		}	
		for(int i=0;i<n;i++)
		{
			if(i<n-1)
				printf("%d ",a[i]);
			else
				printf("%d\n",a[i]);
		}
	}
	return 0;
}