競プロおぼえがき

おぼえがきのためのブログ

yukicoder | No.491 10^9+1と回文

  • 問題

No.491 10^9+1と回文 - yukicoder

  • 概要

1から整数Nまでの数字で 10^{9} + 1の倍数かつ回文になっている数字の個数を求める。

  • 考え方

1から N / (10^{9} + 1)までの数値で回文を求めれば、 10^{9} + 1を掛けたときに 10^{9} + 1の倍数の回文になる。
回文を求めるときは数字を逆にしてくっつけたもの(12 → 1221)みたいな感じでやれば半分の桁の数値までのループで求めれる。

import java.util.Collections;
import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;

public class Main {
    static Scanner in = new Scanner(System.in);
    static Set<Integer> list = new HashSet<>();
    static int max;

    public static void main(String[] args) {
        max = (int) (in.nextLong() / ((long) 1e9 + 1));
        for (int i = 1; i < 10; i++) {
            f(1, i);
        }
        int ans = 0;

        for (int i : list) {
            if (max >= i) {
                ans++;
            }
        }
        System.out.println(ans);
    }

    public static void f(int s, int n) {
        if (s > String.valueOf(max).length()) {
            return;
        }
        if (s == 1) {
            list.add(n);
            f(s + 1, n);
        } else if (s % 2 == 0) {
            String l = String.valueOf(n);
            String r = (new StringBuilder(String.valueOf(n))).reverse().toString();
            list.add(Integer.parseInt(l + r));
            f(s + 1, n);
        } else if (s % 2 == 1) {
            String l = String.valueOf(n);
            String r = (new StringBuilder(String.valueOf(n))).reverse().toString();
            for (int i = 0; i < 10; i++) {
                list.add(Integer.parseInt(l + i + r));
                f(s + 1, n * 10 + i);
            }

        }
    }
}