51Nod--1015 水仙花数

#include <iostream>
#include <cmath>
using namespace std;

// 判断 n 是否为水仙花数
bool jud(int n) {
    int res = n;
    int e = 3;
    if(n >= 1000) {
        e = 4;
    } 
    int sum = 0;
    while(n) {
        sum += pow(n%10, e);
        n /= 10;
    }
    return sum == res;
}

int main() {
    int m;
    cin >> m;

    // 从 m 开始一直找到最小的水仙花数
    for(int i = m; ; i++) {
        if(jud(i)) {
            cout << i << endl;
            break;
        }
    }

    return 0;
}