結果

問題 No.194 フィボナッチ数列の理解(1)
ユーザー 37zigen37zigen
提出日時 2016-05-05 02:14:37
言語 Java21
(openjdk 21)
結果
AC  
実行時間 171 ms / 5,000 ms
コード長 5,866 bytes
コンパイル時間 2,825 ms
コンパイル使用メモリ 81,152 KB
実行使用メモリ 55,780 KB
最終ジャッジ日時 2024-10-05 09:20:26
合計ジャッジ時間 7,822 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 67 ms
50,988 KB
testcase_01 AC 67 ms
50,784 KB
testcase_02 AC 131 ms
52,784 KB
testcase_03 AC 85 ms
52,008 KB
testcase_04 AC 107 ms
52,628 KB
testcase_05 AC 103 ms
52,704 KB
testcase_06 AC 113 ms
53,016 KB
testcase_07 AC 126 ms
53,300 KB
testcase_08 AC 82 ms
52,076 KB
testcase_09 AC 114 ms
52,792 KB
testcase_10 AC 101 ms
53,264 KB
testcase_11 AC 101 ms
52,892 KB
testcase_12 AC 109 ms
53,028 KB
testcase_13 AC 99 ms
52,956 KB
testcase_14 AC 70 ms
50,984 KB
testcase_15 AC 123 ms
52,656 KB
testcase_16 AC 119 ms
52,868 KB
testcase_17 AC 100 ms
52,976 KB
testcase_18 AC 126 ms
53,028 KB
testcase_19 AC 130 ms
53,096 KB
testcase_20 AC 132 ms
55,204 KB
testcase_21 AC 171 ms
55,620 KB
testcase_22 AC 131 ms
54,708 KB
testcase_23 AC 125 ms
55,268 KB
testcase_24 AC 144 ms
55,780 KB
testcase_25 AC 142 ms
55,728 KB
testcase_26 AC 149 ms
55,568 KB
testcase_27 AC 116 ms
54,812 KB
testcase_28 AC 131 ms
55,152 KB
testcase_29 AC 143 ms
54,836 KB
testcase_30 AC 129 ms
52,900 KB
testcase_31 AC 68 ms
50,928 KB
testcase_32 AC 101 ms
52,644 KB
testcase_33 AC 114 ms
53,076 KB
testcase_34 AC 106 ms
52,780 KB
testcase_35 AC 104 ms
53,244 KB
testcase_36 AC 123 ms
52,648 KB
testcase_37 AC 83 ms
52,460 KB
testcase_38 AC 130 ms
53,152 KB
testcase_39 AC 105 ms
52,672 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.IOException;
import java.util.Arrays;
import java.util.InputMismatchException;

public class Main {
    long MOD = 1_000_000_000 + 7;

    class Node {
        long value;
        Node next;

        Node(long value, Node next) {
            this.value = value;
            this.next = next;
        }
    }

    void test1(int n, long k, long[] a) {
        Node first = null;
        Node last = null;

        Node p = null;
        long sum = 0;
        for (int i = 0; i < n; i++) {
            Node tmp = new Node(a[i], null);
            if (first == null) {
                first = tmp;
                p = tmp;
            } else {
                p.next = tmp;
                p = tmp;
            }
            sum += a[i];
            sum %= MOD;
        }
        last = p;

        long tmp = sum;
        for (int i = n; i < k; i++) {
            Node node = new Node(tmp, null);

            sum += tmp;

            tmp -= first.value;
            first = first.next;
            last.next = node;
            last = last.next;
            tmp += last.value;

            tmp %= MOD;
            tmp += MOD;
            tmp %= MOD;
            sum %= MOD;
            sum += MOD;
            sum %= MOD;
        }
        System.out.println(last.value + " " + sum);
    }

    void test2(int n, long k, long[] a) {
        long[][] mat = new long[n + 1][n + 1];

        Arrays.fill(mat[0], 1);
        Arrays.fill(mat[1], 1);
        mat[1][0] = 0;
        for (int i = 1; i < n; i++) {
            mat[i + 1][i] = 1;
        }

        long[][] tmp = power(mat, k - n);
        long[] x = new long[n + 1];
        for (int i = 0; i < n; i++) {
            x[0] += a[i];
            x[i + 1] = a[n - i - 1];
        }
        long ans1 = 0, ans2 = 0;
        for (int i = 0; i < n + 1; i++) {
            ans2 += tmp[0][i] * x[i];
            ans1 += tmp[1][i] * x[i];
            ans1 %= MOD;
            ans2 %= MOD;
        }

        System.out.println(ans1 + " " + ans2);
    }

    long[][] power(long[][] a, long k) {
        int n = a.length;
        long[][] res = new long[n][n];
        long[][] tmp = new long[n][n];

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                tmp[i][j] = a[i][j];
            }
            res[i][i] = 1;
        }

        while (0 < k) {
            if (k % 2 == 1) {
                res = mul(res, tmp);
            }
            tmp = mul(tmp, tmp);
            k >>= 1;
        }

        return res;
    }

    long[][] mul(long[][] a, long[][] b) {
        int n = a.length;
        long[][] res = new long[n][n];

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                res[i][j] = 0;
                for (int k = 0; k < n; k++) {
                    res[i][j] += a[i][k] * b[k][j];
                    res[i][j] %= MOD;
                }
            }
        }

        return res;
    }

    void run() {
        MyScanner sc = new MyScanner();

        int n = sc.nextInt();
        long k = sc.nextLong();
        long[] a = sc.nextLongArray(n);

        if (n <= 10000 && k <= 1000000) {
            test1(n, k, a);
        } else {
            test2(n, k, a);
        }
    }

    public static void main(String[] args) {
        new Main().run();
    }

    public void mapDebug(int[][] a) {
        System.out.println("--------map display---------");
        for (int i = 0; i < a.length; i++) {
            for (int j = 0; j < a[i].length; j++) {
                System.out.printf("%3d ", a[i][j]);
            }
            System.out.println();
        }
        System.out.println("----------------------------" + '\n');
    }

    class MyScanner {
        int read() {
            try {
                return System.in.read();
            } catch (IOException e) {
                throw new InputMismatchException();
            }
        }

        boolean isSpaceChar(int c) {
            return c == ' ' || c == '\n' || c == '\r' || c == '\t' || c == -1;
        }

        boolean isEndline(int c) {
            return c == '\n' || c == '\r' || c == -1;
        }

        int nextInt() {
            return Integer.parseInt(next());
        }

        int[] nextIntArray(int n) {
            int[] array = new int[n];
            for (int i = 0; i < n; i++)
                array[i] = nextInt();
            return array;
        }

        long nextLong() {
            return Long.parseLong(next());
        }

        long[] nextLongArray(int n) {
            long[] array = new long[n];
            for (int i = 0; i < n; i++)
                array[i] = nextLong();
            return array;
        }

        double nextDouble() {
            return Double.parseDouble(next());
        }

        double[] nextDoubleArray(int n) {
            double[] array = new double[n];
            for (int i = 0; i < n; i++)
                array[i] = nextDouble();
            return array;
        }

        String next() {
            int c = read();
            while (isSpaceChar(c))
                c = read();
            StringBuilder res = new StringBuilder();
            do {
                res.appendCodePoint(c);
                c = read();
            } while (!isSpaceChar(c));
            return res.toString();
        }

        String[] nextStringArray(int n) {
            String[] array = new String[n];
            for (int i = 0; i < n; i++)
                array[i] = next();

            return array;
        }

        String nextLine() {
            int c = read();
            while (isEndline(c))
                c = read();
            StringBuilder res = new StringBuilder();
            do {
                res.appendCodePoint(c);
                c = read();
            } while (!isEndline(c));
            return res.toString();
        }
    }
}
0