BNY Mellon Coding Questions and Answers

#include<bits/stdc++.h>

using namespace std;

bool isPrime(int n) {
    if (n <= 1)
        return false;
    if (n <= 3)
        return true;
    if (n % 2 == 0 || n % 3 == 0)
        return false;
    for (int i = 5; i * i <= n; i += 6)
        if ((n % i == 0) || (n % (i + 2) == 0))
            return false;
    return true;
}

int solve(int n, int k) {
    int c = 0;
    vector list;

    // Find k numbers greater than n with the property that the sum of their digits is prime
    for (int i = n + 1; c < k; i++) {
        int temp = i;
        int sum = 0;
        while (temp != 0) {
            sum += temp % 10;
            temp /= 10;
        }
        if (isPrime(sum)) {
            list.push_back(i);
            c++;
        }
    }
    return list[k - 1];
}

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

    int k;
    cin >> k;

    cout << solve(n, k);

    return 0;
}

import java.util.*;
class Main {
    public static boolean isPrime(int n) {
        if (n <= 1)
            return false;
        if (n <= 3)
            return true;
        if (n % 2 == 0 || n % 3 == 0)
            return false;
        for (int i = 5; i * i <= n; i = i + 6)
            if ((n % i == 0) || (n % (i + 2) == 0))
                return false;
        return true;
    }
    public static int solve(int n, int k) {
        int c = 0;
        ArrayList < Integer > list = new ArrayList < > ();
        for (int i = n + 1; c < k; i++) {
            int temp = i;
            int sum = 0;
            while (temp != 0) {
                sum = sum + temp % 10;
                temp = temp / 10;
            }
            if (isPrime(sum)) {
                list.add(i);
                c++;
            }
        }
        return list.get(k - 1);
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        System.out.println(solve(n, k));
    }
}

import math

def isPrime(n):
    if n <= 1:
        return False
    if n <= 3:
        return True
    if n % 2 == 0 or n % 3 == 0:
        return False
    for i in range(5, int(math.sqrt(n)) + 1, 6):
        if n % i == 0 or n % (i + 2) == 0:
            return False
    return True


def solve(n, k):
    c = 0
    list = []
    for i in range(n + 1, 1000000000):
        temp = i
        sum = 0
        while temp != 0:
            sum = sum + temp % 10
            temp = temp // 10
        if isPrime(sum):
            list.append(i)
            c += 1
        if c == k:
            break
    return list[k - 1]


n, k = map(int, input().split())
print(solve(n, k))