L1-027 出租 (20分)

L1-027 出租 (20分)

题目

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

一时间网上一片求救声，急问这个怎么破。其实这段代码很简单，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};

思路

两个数组，第一个数组只保存号码中不重复的数字，并从大到小排列；第二个数组保存第一个数组的反向映射（vis[arr[i]] = i;）。最后输出第一个数组与原号码通过第二个数组的映射后的数组即可。具体实现见代码。

代码

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PI acos(-1)
using namespace std;
typedef long long ll;
const int N = 2e5 + 9;
const ll mod = 1e9 + 7;

char str[N];
int vis[N];
int arr[N];

bool cmp(int a, int b)
{
    return a > b;
}

int main()
{
    cin >> str;
    int n = strlen(str);
    int cnt = 0;
    for(int i = 0; i < n; i++)
    {
        int x = str[i] - '0';
        if(!vis[x])
        {
            cnt++;
            arr[cnt] = x;
            vis[x] = cnt;
        }
    }
    sort(arr + 1, arr + cnt + 1, cmp);
    for(int i = 1; i <= cnt; i++)
    {
        vis[arr[i]] = i;
    }
    cout << "int[] arr = new int[]{";
    for(int i = 1; i <= cnt; i++)
    {
        cout << arr[i];
        if(i != cnt)
            cout << ",";
    }
    cout << "};" << endl;
    cout << "int[] index = new int[]{";
    for(int i = 0; i < n; i++)
    {
        cout << vis[str[i] - '0'] - 1;
        if(i != n - 1)
            cout << ",";
    }
    cout << "};" << endl;
    return 0;
}