Advertisement
Guest User

Untitled

a guest
Feb 21st, 2019
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.78 KB | None | 0 0
  1. import java.util.Scanner;
  2.  
  3. public class Solution {
  4. private static boolean isPentagonal(long x) {
  5. long n = (long) ((1 + Math.sqrt(24 * x + 1)) / 6);
  6. long p_n = n * (3 * n - 1) / 2;
  7. return p_n == x;
  8. }
  9.  
  10. public static void main(String[] args) {
  11. try (Scanner sc = new Scanner(System.in)) {
  12. int N = sc.nextInt();
  13. int K = sc.nextInt();
  14. for(long n = K + 1; n <= N; n++) {
  15. long p_n = n * (3 * n - 1) / 2;
  16. long p_x = (n - K) * (3 * (n - K) - 1) / 2;
  17. long sum = p_n + p_x;
  18. long difference = p_n - p_x;
  19. if(isPentagonal(sum) || isPentagonal(difference)) {
  20. System.out.println(p_n);
  21. }
  22. }
  23. }
  24. }
  25. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement