奶牛生牛问题

2021-10-13 17:35:35 阅读：163 来源： 互联网

标签：生牛 int long 问题 length static result 奶牛母牛

假设农场中成熟的母牛每年只会生 1 头小母牛，并且永远不会死。第一年农场中有一只成熟的母牛，从第二年开始，母牛开始生小母牛。每只小母牛 3 年之后成熟又可以生小母牛。给定整数 n，求出 n 年后牛的数量。

f(1) = 1
f(2) = 2
f(3) = 3
f(n) = f(n - 1) + f(n - 3) (n > 3)
f(n) = (f(3), f(2), f(1)) (x ^ (n - 3))
所以有，
(f(4), f(3), f(2)) = (f(3), f(2), f(1)) （x ^ 1）
(f(5), f(4), f(3)) = (f(3), f(2), f(1)) （x ^ 2）
(f(6), f(5), f(4)) = (f(3), f(2), f(1)) （x ^ 3）
得矩阵
{1, 1, 0},
{0, 0, 1}.
{1, 0, 0}

import java.util.Scanner;

public class Main {

    private static final int MOD = 1000000007;

    private static long[][] multiply(long[][] a, long[][] b) {
        long[][] result = new long[a.length][b[0].length];

        for (int i = 0; i < a.length; ++i) {
            for (int j = 0; j < b[0].length; ++j) {
                for (int k = 0; k < a[0].length; ++k) {
                    result[i][j] = (result[i][j] + a[i][k] * b[k][j]) % MOD;
                }
            }
        }
        return result;
    }

    private static long solve(long n) {
        if (n <= 3) {
            return n;
        }

        n -= 3;

        long[][] result = {
                {3, 2, 1}
        };
        long[][] base = {
                {1, 1, 0},
                {0, 0, 1},
                {1, 0, 0}
        };

        while (n > 0) {
            if ((n & 1) != 0) {
                result = multiply(result, base);
            }
            n >>= 1;
            base = multiply(base, base);
        }
        return result[0][0];
    }

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while (in.hasNext()) {
            long n = in.nextLong();
            System.out.println(solve(n));
        }

    }
}

标签：生牛,int,long,问题,length,static,result,奶牛,母牛
来源： https://www.cnblogs.com/tianyiya/p/15403167.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

奶牛生牛问题