PAT L1-027 出租

https://pintia.cn/problem-sets/994805046380707840/problems/994805107638517760

下面是新浪微博上曾经很火的一张图：

一时间网上一片求救声，急问这个怎么破。其实这段代码很简单，index数组就是arr数组的下标，index[0]=2 对应 arr[2]=1，index[1]=0 对应 arr[0]=8，index[2]=3对应 arr[3]=0，以此类推…… 很容易得到电话号码是18013820100。

本题要求你编写一个程序，为任何一个电话号码生成这段代码 —— 事实上，只要生成最前面两行就可以了，后面内容是不变的。

输入格式：

输入在一行中给出一个由11位数字组成的手机号码。

输出格式：

为输入的号码生成代码的前两行，其中arr中的数字必须按递减顺序给出。

输入样例：

18013820100

输出样例：

int[] arr = new int[]{8,3,2,1,0};
int[] index = new int[]{3,0,4,3,1,0,2,4,3,4,4};

代码：

#include <bits/stdc++.h>
using namespace std;
 
const int maxn = 1e5 + 10;
string s;
int pos[maxn];
 
vector<int> num, ind;
map<int, int> mp;
 
int main() {
    cin >> s;
    int len = s.length();
    for(int i = 0; i < len; i ++) {
        if(mp[s[i] - '0'] == 0) {
            mp[s[i] - '0'] = 1;
            num.push_back(s[i] - '0');
        } else continue;
    }
 
    sort(num.rbegin(), num.rend());
    for(int i = 0; i < num.size(); i ++)
        pos[num[i]] = i;
 
    for(int i = 0; i <len; i ++)
        ind.push_back(pos[s[i] - '0']);
 
    printf("int[] arr = new int[]{");
    for(int i = 0; i < num.size(); i ++)
        printf("%d%s", num[i], i != num.size() - 1 ? "," : "};\n");
    printf("int[] index = new int[]{");
    for(int i = 0; i < ind.size(); i ++)
        printf("%d%s", ind[i], i != ind.size() - 1 ? "," : "};");
    return 0;
}

$O(len)$