fzu 2184 逆序数还原

FZU 2184 逆序数还原

http://acm.fzu.edu.cn/problem.php?pid=2184

 Problem 2184 逆序数还原

Accept: 94    Submit: 165
Time Limit: 1000 mSec    Memory Limit : 32768 KB

 Problem Description

有一段时间Eric对逆序数充满了兴趣，于是他开始求解许多数列的逆序数（对于由1...n构成的一种排列数组a，逆序数即为满足i<j,ai>aj的数字对数），但是某天他发现自己遗失了原来的数列，只留下之前计算过程中留下的各个数字对应的逆序数，现在请你帮他还原出原序列。

 Input

数据有多组，请处理到文件结尾。

每组数据第一行为一个整数N（1<=N<=1000），表示该序列的数字个数。

第二行为N个整数，第i个数字表示排在ai之后比ai小的数字个数。

 Output

输出为一行N个整数，表示原数列。

 Sample Input

5

2 0 1 0 0

 Sample Output

3 1 4 2 5

 

分析：

首先将数组初始化为1……N，想到后面还会有“删除”操作，想到用STL 里的vector容器。

AC代码：

#include <stdio.h>
#include <iostream>
#include <stdlib.h>
#include <algorithm>
#include <string.h>
#include <string>
#include <math.h>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>

using namespace std;

const int INF = 0x3f3f3f3f;
const int MAX = 1000 + 10;
const double eps = 1e-7;
const double PI = acos(-1.0);

vector<int> v;
int a[MAX];

int main()
{
    int n;
    while(~scanf("%d", &n))
    {
        v.clear();
        int i , x;
        for(i = 1; i <= n; i++)
            v.push_back(i);
        vector<int>::iterator it;
        for(i = 1; i <= n; i++)
        {
            scanf("%d", &x);
            a[i] = v[x];
            it = v.begin() + x;
            v.erase(it);
        }
        printf("%d", a[1]);
        for(i = 2; i <= n; i++)
            printf(" %d", a[i]);
        puts("");
    }
    return 0;
}