Kyselyt

import java.util.*;
/*
 * Datatähti 2018
 * Tuomas Karjalainen, Nurmeksen lukio
 */
public class Kyselyt {
  public static void main(String[] args) {
    Scanner input = new Scanner(System.in);
    int queries = Integer.parseInt(input.nextLine());
    while (queries-- > 0) {
      long k = Long.parseLong(input.nextLine());
      System.out.println(query(k, true));
      System.out.println(query(k, false));
    }
  }

  private static char query(long k, boolean new_method) {
    long index = 1;
    long currentNumber = 1;
    long numberLength, numberCountWithLength;
    if (new_method) {
      while ((numberLength = numberLength(currentNumber)) > 0 &&
            (numberCountWithLength = numberCountWithLength(numberLength)) > 0 &&
            (index + numberCountWithLength * numberLength) <= k) {
        index += numberCountWithLength * numberLength;
        currentNumber += numberCountWithLength;
      }
    }
    while ((numberLength = numberLength(currentNumber)) > 0 &&
          index + numberLength <= k) {

      index += numberLength;
      currentNumber++;
    }
    return Long.toString(currentNumber).charAt((int)(k - index));
  }

  private static long numberLength(long number) {
    long length = 0;
    while (number > 0) {
      number /= 10;
      length++;
    }
    return length;
  }

  private static long numberCountWithLength(long length) {
    long count = 9;
    while (--length > 0) {
      count *= 10;
    }
    return count;
  }
}

Test 1

Group: 1

Verdict: WRONG ANSWER

Test 2

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED