【华为OD机试真题 C++】水仙花数 【2022 Q4 | 100分】

■ 题目描述

【水仙花数】

所谓水仙花数，是指一个n位的正整数，其各位数字的n次方和等于该数本身。

例如153是水仙花数，153是一个3位数，并且153 = 1^3 + 5^3 + 3^3。

输入描述

第一行输入一个整数n，表示一个n位的正整数。n在3到7之间，包含3和7。

第二行输入一个正整数m，表示需要返回第m个水仙花数。

输出描述

返回长度是n的第m个水仙花数。个数从0开始编号。

若m大于水仙花数的个数，返回最后一个水仙花数和m的乘积。

若输入不合法，返回-1。

示例1 输入输出示例仅供调试，后台判题数据一般不包含示例

输入

3 0

输出

153

说明

153是第一个水仙花数

示例2 输入输出示例仅供调试，后台判题数据一般不包含示例

输入

9

1

输出

-1

说明

9超出范围

C++代码实现：

#include <bits/stdc++.h>

using namespace std;

static int getCalcuteNum(int a, int b)
{
    int result = 1;
    for (int i = 0; i < b; i++) {
        result *= a;
    }
    return result;
}

bool isRightNum(int num, int n)
{
    bool result = false;
    vector<int> data;
    for (char c: to_string(num))
        data.push_back(c - 0);
    int sum = 0;
    for (int i = 0; i < n; i++) {
        sum += getCalcuteNum(data[i], n);
    }
    if (sum == num) {
        result = true;
    }
    return result;
}

int main(int argc, const char *argv[])
{
    int n;
    cin >> n;
    if (n >= 3 && n <= 7) {
        int m;
        cin >> m;
        int temp = 1;
        for (int i = 0; i < n; i++) {
            temp *= 10;
        }
        int min = temp / 10;
        int max = temp - 1;
        unordered_map<int, int> data;
        int count = 0;
        int lastNum = 0;
        for (int i = min; i <= max; ++i) {
            if (isRightNum(i, n)) {
                data.insert({count++, i});
                if (i > lastNum) {
                    lastNum = i;
                }
            }
        }
        if (data.count(m)) {
            cout << data[m] << endl;
        } else {
            cout << lastNum * m << endl;
        }
    } else {
        cout << -1 << endl;
    }
    return 0;
}