結果

問題 No.194 フィボナッチ数列の理解(1)
ユーザー eitahoeitaho
提出日時 2015-04-27 20:52:43
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 1,549 ms / 5,000 ms
コード長 5,758 bytes
コンパイル時間 2,554 ms
コンパイル使用メモリ 110,132 KB
実行使用メモリ 33,580 KB
最終ジャッジ日時 2023-09-18 13:56:20
合計ジャッジ時間 22,308 ms
ジャッジサーバーID
(参考情報)
judge14 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 68 ms
25,168 KB
testcase_01 AC 68 ms
23,240 KB
testcase_02 AC 1,549 ms
31,580 KB
testcase_03 AC 170 ms
27,348 KB
testcase_04 AC 601 ms
29,544 KB
testcase_05 AC 448 ms
31,356 KB
testcase_06 AC 539 ms
33,508 KB
testcase_07 AC 892 ms
27,408 KB
testcase_08 AC 143 ms
27,340 KB
testcase_09 AC 653 ms
31,460 KB
testcase_10 AC 305 ms
29,560 KB
testcase_11 AC 297 ms
31,436 KB
testcase_12 AC 473 ms
31,600 KB
testcase_13 AC 214 ms
31,384 KB
testcase_14 AC 95 ms
25,280 KB
testcase_15 AC 1,026 ms
29,556 KB
testcase_16 AC 866 ms
29,492 KB
testcase_17 AC 276 ms
31,380 KB
testcase_18 AC 952 ms
29,364 KB
testcase_19 AC 1,368 ms
29,572 KB
testcase_20 AC 93 ms
27,296 KB
testcase_21 AC 105 ms
27,816 KB
testcase_22 AC 96 ms
29,180 KB
testcase_23 AC 80 ms
26,068 KB
testcase_24 AC 91 ms
30,216 KB
testcase_25 AC 91 ms
28,176 KB
testcase_26 AC 92 ms
30,412 KB
testcase_27 AC 90 ms
28,892 KB
testcase_28 AC 81 ms
26,064 KB
testcase_29 AC 101 ms
29,012 KB
testcase_30 AC 1,414 ms
33,580 KB
testcase_31 AC 71 ms
25,396 KB
testcase_32 AC 441 ms
31,472 KB
testcase_33 AC 583 ms
29,492 KB
testcase_34 AC 476 ms
29,428 KB
testcase_35 AC 442 ms
29,428 KB
testcase_36 AC 1,116 ms
31,480 KB
testcase_37 AC 139 ms
27,356 KB
testcase_38 AC 1,161 ms
31,508 KB
testcase_39 AC 508 ms
29,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff #

using System;
using System.IO;
using System.Linq;
using System.Text;
using System.Collections.Generic;
using System.Diagnostics;
using System.Numerics;
using Enu = System.Linq.Enumerable;

class Program
{
    static readonly int SmallN = (int)1e6;
    static readonly int Mod = (int)1e9 + 7;

    public void Solve()
    {
        int L = Reader.Int();
        long N = Reader.Long();
        var F = new[] { 0 }.Concat(Reader.IntArray(L)).Concat(new int[SmallN]).ToArray();
        var S = new int[SmallN + 1];
        for (int i = 1; i <= L; i++) S[i] = (F[i] + S[i - 1]) % Mod;

        if (N <= SmallN)
        {
            for (int i = L + 1; i <= N; i++)
            {
                F[i] = (S[i - 1] - S[i - L - 1] + Mod) % Mod;
                S[i] = (F[i] + S[i - 1]) % Mod;
            }
            Console.WriteLine(F[N] + " " + S[N]);
        }
        else
        {
            var M = new Matrix(L, L, Mod);
            for (int c = 0; c < L; c++) M[0, c] = 1;
            for (int r = 1; r < L; r++) M[r, r - 1] = 1;
            var Fmat = new Matrix(F.Skip(1).Take(L).Reverse().ToArray(), Mod);
            long f = (M.Power(N - L) * Fmat)[0, 0];
            long s = (S[L] + (Rec(M, N - L) * Fmat)[0, 0]) % Mod;
            Console.WriteLine(f + " " + s);
        }
    }

    Matrix Rec(Matrix A, long pow)
    {
        if (pow == 1) return A;
        var halfSum = Rec(A, pow / 2);
        var res = A.Power(pow / 2) * halfSum;
        res += halfSum;
        if (pow % 2 == 1) res += A.Power(pow);
        return res;
    }

    class Matrix
    {
        public readonly int H, W;
        public readonly long Mod;
        private readonly long[][] M;

        public long this[int r, int c]
        {
            get { return M[r][c]; }
            set { M[r][c] = (value % Mod + Mod) % Mod; }
        }
        public Matrix(int h, int w, long mod)
        {
            H = h; W = w; Mod = mod; M = new long[h][];
            for (int r = 0; r < H; r++) M[r] = new long[w];
        }
        public Matrix(long[] vec, long mod)
        {
            H = vec.Length; W = 1; Mod = mod; M = new long[H][];
            for (int r = 0; r < H; r++) { M[r] = new long[1]; this[r, 0] = vec[r]; }
        }
        public Matrix(int[] vec, long mod)
        {
            H = vec.Length; W = 1; Mod = mod; M = new long[H][];
            for (int r = 0; r < H; r++) { M[r] = new long[1]; this[r, 0] = vec[r]; }
        }
        public static Matrix operator +(Matrix A, Matrix B)
        {
            Debug.Assert(A.H == B.H && A.W == B.W && A.Mod == B.Mod);
            var res = A.Clone();
            for (int r = 0; r < res.H; r++)
                for (int c = 0; c < res.W; c++)
                    Add(ref res.M[r][c], B.M[r][c], res.Mod);
            return res;
        }
        public static Matrix operator *(Matrix A, Matrix B)
        {
            Debug.Assert(A.W == B.H && A.Mod == B.Mod);
            long mod = A.Mod;
            var res = new Matrix(A.H, B.W, mod);
            for (int r = 0; r < res.H; r++)
                for (int i = 0; i < A.W; i++)
                    for (int c = 0; c < res.W; c++)
                        Add(ref res.M[r][c], A.M[r][i] * B.M[i][c] % mod, mod);
            return res;
        }
        public Matrix Power(long pow)
        {
            Debug.Assert(H == W);
            var A = Clone();
            var res = new Matrix(H, W, Mod);
            for (int i = 0; i < H; i++) res[i, i] = 1;
            while (pow > 0)
            {
                if ((pow & 1) == 1) res = res * A;
                A = A * A;
                pow >>= 1;
            }
            return res;
        }
        private static void Add(ref long a, long b, long mod)
        {
            if ((a += b) >= mod) a -= mod;
            if (a < 0) a += mod;
        }
        public Matrix Clone()
        {
            var res = new Matrix(H, W, Mod);
            for (int r = 0; r < H; r++) Array.Copy(M[r], res.M[r], res.W);
            return res;
        }
    }
}


class Entry { static void Main() { new Program().Solve(); } }
class Reader
{
    private static TextReader reader = Console.In;
    private static readonly char[] separator = { ' ' };
    private static readonly StringSplitOptions op = StringSplitOptions.RemoveEmptyEntries;
    private static string[] A = new string[0];
    private static int i;
    private static void Init() { A = new string[0]; }
    public static void Set(TextReader r) { reader = r; Init(); }
    public static void Set(string file) { reader = new StreamReader(file); Init(); }
    public static bool HasNext() { return CheckNext(); }
    public static string String() { return Next(); }
    public static int Int() { return int.Parse(Next()); }
    public static long Long() { return long.Parse(Next()); }
    public static double Double() { return double.Parse(Next()); }
    public static int[] IntLine() { return Array.ConvertAll(Split(Line()), int.Parse); }
    public static int[] IntArray(int N) { return Enu.Range(0, N).Select(i => Int()).ToArray(); }
    public static int[][] IntTable(int H) { return Enu.Range(0, H).Select(i => IntLine()).ToArray(); }
    public static string[] StringArray(int N) { return Enu.Range(0, N).Select(i => Line()).ToArray(); }
    public static string Line() { return reader.ReadLine().Trim(); }
    private static string[] Split(string s) { return s.Split(separator, op); }
    private static string Next() { CheckNext(); return A[i++]; }
    private static bool CheckNext()
    {
        if (i < A.Length) return true;
        string line = reader.ReadLine();
        if (line == null) return false;
        if (line == "") return CheckNext();
        A = Split(line);
        i = 0;
        return true;
    }
}
0