C题矩阵快速幂

数列求值

https://ac.nowcoder.com/acm/contest/6631/C

显然1e18的复杂度O(n)的dp做法是不可取的
那就矩阵快速幂
如果没有了解可以看下B站OTTFF的视频
就是一个模板题,的确就是wlp大佬说的会的就会

/*
a0=0 a1=1
ai=b*ai-1+ c*ai-2
[0 1]^n * [a0] = [an ]
[c b]     [a1] = [an+1]
*/
long long mod=1e9+7;
const int M=2;
class Ma
{
public:
    long long a[M][M];
    Ma()
    {
        memset(a,0,sizeof(a));
    }
    void become_unit_matrix()
    {
        //设置成单位矩阵
        for (int i=0;i<M;++i)
            for (int j=0;j<M;++j)
                a[i][j]= i==j ? 1 : 0;
    }
    void show()
    {
        cout << "----" << endl;
        cout << "Matrix is " << M << '*' << M << endl;
        for (int i=0;i<M;++i)
        {
            for (int j=0;j<M;++j)
                cout << a[i][j] << ' ';
            cout << endl;
        }
        cout << "----" << endl;
    }
    Ma operator*(const Ma &B) const
    {
        //重载两个矩阵相乘
        Ma ans;
        for (int i=0;i<M;++i)
            for (int j=0;j<M;++j)
                for (int k=0;k<M;++k)
                {
                    ans.a[i][j]+=(((this->a[i][k])%mod)*((B.a[k][j])%mod))%mod;
                    ans.a[i][j]%=mod;
                }
        return ans;
    }
    Ma operator^(long long n) const
    {
        //重载幂次运算,并结合快速幂
        Ma ans;
        //单位矩阵乘任何矩阵还是任何矩阵本身
        ans.become_unit_matrix();
        Ma A=*this;
        while(n!=0)
        {
            if ((n&1)==1)
                ans=ans*A;
            A=A*A;
            n>>=1;
        }
        return ans;
    }
};
class Solution {
public:
    long long nthElement(long long n, long long b, long long c) {
        Ma A;
        A.a[0][0]=0;
        A.a[0][1]=1;
        A.a[1][0]=c;
        A.a[1][1]=b;
        Ma F;
        F.a[0][0]=0;
        F.a[1][0]=1;
        Ma ans=(A^n)*F;
        return ans.a[0][0];
    }
};
全部评论

相关推荐

找个工作&nbsp;学历是要卡的&nbsp;要求是高的&nbsp;技能不足是真的&nbsp;实习经验是0的&nbsp;简历无处可写是事实的&nbsp;钱不好赚是真的&nbsp;想躺平又不敢躺&nbsp;也不甘心躺&nbsp;怕自己的灵感和才华被掩埋甚至从未被自己发现&nbsp;又质疑自己是否真正有才华
码农索隆:你现在啊,你心里都明白咋回事,但是你没办法改变现状,一想到未来,你又没有信心狠下心来在当下努力。 得走出这种状态,不能一直困在那里面,哪不行就去提升哪,你一动不动那指定改变不了未来,动起来,积少成多才能越来越好
点赞 评论 收藏
分享
06-07 17:17
嘉兴学院 教师
心爱的idea:你孩
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务