結果
問題 | No.1731 Product of Subsequence |
ユーザー |
![]() |
提出日時 | 2023-07-19 09:49:31 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 757 ms / 2,000 ms |
コード長 | 3,311 bytes |
コンパイル時間 | 2,431 ms |
コンパイル使用メモリ | 89,436 KB |
実行使用メモリ | 64,852 KB |
最終ジャッジ日時 | 2024-09-19 08:28:28 |
合計ジャッジ時間 | 13,376 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 31 |
ソースコード
import java.io.*;import java.util.*;import java.util.stream.*;public class Main {static int[] values;static int[][] dp;static final int MOD = 1000000007;static int[][] matrix;public static void main(String[] args) throws Exception {Scanner sc = new Scanner();int n = sc.nextInt();int k = sc.nextInt();if (k == 1) {System.out.println(pow(2, n) - 1);return;}TreeMap<Integer, Integer> idxes = new TreeMap<>();for (int i = 1; i <= Math.sqrt(k); i++) {if (k % i == 0) {idxes.put(i, null);idxes.put(k / i, null);}}int size = 0;for (int x : idxes.keySet()) {idxes.put(x, size++);}matrix = new int[size][size];for (int x : idxes.keySet()) {for (int y : idxes.keySet()) {matrix[idxes.get(x)][idxes.get(y)] = idxes.get((int)(getGCD((long)x * y, k)));}}values = new int[n];for (int i = 0; i < n; i++) {values[i] = idxes.get((int)getGCD(k, sc.nextLong()));}dp = new int[n][size];for (int[] arr : dp) {Arrays.fill(arr, -1);}System.out.println(dfw(n - 1, 0));}static long pow(long x, int p) {if (p == 0) {return 1;} else if (p % 2 == 0) {return pow(x * x % MOD, p / 2);} else {return pow(x, p - 1) * x % MOD;}}static long getGCD(long x, long y) {if (y == 0) {return x;} else {return getGCD(y, x % y);}}static int dfw(int idx, int v) {if (v == matrix.length - 1) {return (int)pow(2, idx + 1);}if (idx < 0) {return 0;}if (dp[idx][v] < 0) {dp[idx][v] = (dfw(idx - 1, v) + dfw(idx - 1, matrix[v][values[idx]])) % MOD;}return dp[idx][v];}}class Utilities {static String arrayToLineString(Object[] arr) {return Arrays.stream(arr).map(x -> x.toString()).collect(Collectors.joining("\n"));}static String arrayToLineString(int[] arr) {return String.join("\n", Arrays.stream(arr).mapToObj(String::valueOf).toArray(String[]::new));}}class Scanner {BufferedReader br = new BufferedReader(new InputStreamReader(System.in));StringTokenizer st = new StringTokenizer("");StringBuilder sb = new StringBuilder();public Scanner() throws Exception {}public int nextInt() throws Exception {return Integer.parseInt(next());}public long nextLong() throws Exception {return Long.parseLong(next());}public double nextDouble() throws Exception {return Double.parseDouble(next());}public int[] nextIntArray() throws Exception {return Stream.of(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();}public String next() throws Exception {while (!st.hasMoreTokens()) {st = new StringTokenizer(br.readLine());}return st.nextToken();}}