Hi
Have sample tests work fine, but the last test still fails.
Could you give me some hints, pleaSe?
Thank you!
package com.codegym.task.task20.task2025;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
/*
Number algorithms
*/
public class Solution {
public static long[] getNumbers(long N) {
ArrayList<Long> arrayList = new ArrayList<>();
long[] result = null;
HashMap<String,Integer> cache = new HashMap<>();
for (long i = 1; i<N;i++) {
long sum = 0;
String num = String.valueOf(i);
int digitsCount = num.length();
for (int indx = 0; indx<digitsCount; indx++) {
byte digit = Byte.parseByte(String.valueOf(num.charAt(indx)));
String key = digit + "-" + digitsCount;
if (cache.containsKey(key)) {
sum += cache.get(key);
} else {
cache.put(key, pow(digit, digitsCount));
sum += cache.get(key);
}
}
if (sum == i) {
arrayList.add(i);
}
}
result = new long[arrayList.size()];
for (int i =0; i<arrayList.size();i++) {
result[i] = arrayList.get(i);
}
return result;
}
public static int pow(int a, int b) {
int result = 1;
for (int i=1;i<=b;i++) {
result *= a;
}
return result;
}
public static void main(String[] args) {
long a = System.currentTimeMillis();
System.out.println(Arrays.toString(getNumbers(1000)));
long b = System.currentTimeMillis();
System.out.println("memory " + (Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory()) / (8 * 1024));
System.out.println("time = " + (b - a) / 1000);
a = System.currentTimeMillis();
System.out.println(Arrays.toString(getNumbers(1000000)));
b = System.currentTimeMillis();
System.out.println("memory " + (Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory()) / (8 * 1024));
System.out.println("time = " + (b - a) / 1000);
}
}