sdnu1313 Chess (找规律 斐波那契+矩阵快速幂模板)

Description

小X有一个1*n的跳棋棋盘。今天小X要去参加比赛,他希望所有的跳棋排成他希望的队形(即在指定的格子上有棋子)来给他加油。小X只能在棋盘的第1个格或第2个格放置棋子,而其他的格子只能通过跳棋的跳跃到达。当且仅当第i格有棋子,i+1格有棋子,i+2格为空的时候,i格上的棋子能够跳到i+2格上,与此同时i+1格子上的棋子会消失。同理,当且仅当第i格有棋子,i-1格有棋子,i-2格为空的时候,i格上的棋子能够跳到i-2格上,与此同时i-1格子上的棋子会消失。

现在告诉你小X希望那些格子上需要有棋子,你的任务是求出最少需要在棋盘上放置多少个棋子才能做到。

 

由于这个答案可能过大,为了简化问题,你只需要输出最终答案模268435456的余数。

Input

第一行2个整数:n,m。n表示棋盘的大小,m表示希望放置棋子的格子个数。2<n<=100000,m<=n。所有的a∈[3,n],保证按非减序给出。

 

接下来m行,每行一个整数a,表示在第a个格子上需要放置一枚棋子。

Output

一个整数,最少需要放置的棋子数量 mod 268435456。

Sample Input

4 2
3 
4

Sample Output

5

不难发现f(n)=f(n-1)+f(n-2)  1e5的数据 递推会超时的 正好学一下矩阵快速幂

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2;
const int MOD=268435456;
struct mat
{
    ll a[N][N];
};
mat mat_mul(mat x,mat y)
{
    mat res;
    memset(res.a,0,sizeof(res.a));
    for(int i=0; i<2; i++)
        for(int j=0; j<2; j++)
            for(int k=0; k<2; k++)
                res.a[i][j]=(res.a[i][j]+x.a[i][k]*y.a[k][j])%MOD;
    return res;
}
long long mat_pow(ll n)
{
    mat c,res;
    c.a[0][0]=c.a[0][1]=c.a[1][0]=1;
    c.a[1][1]=0;
    memset(res.a,0,sizeof(res.a));
    for(int i=0; i<2; i++)
        res.a[i][i]=1;
    while(n)
    {
        if(n&1)
            res=mat_mul(res,c);
        c=mat_mul(c,c);
        n=n>>1;
    }
    return res.a[0][1];
}
int main()
{
    ll n;
    int a,b;
    scanf("%d%d",&a,&b);
    long long ans=0;
    while(b--)
    {
        scanf("%lld",&n);
        long long tmp=mat_pow(n);
        ans=(ans%MOD+tmp%MOD)%MOD;
    }
    cout<<ans<<'\n';
    return 0;
}

 

全部评论

相关推荐

11-18 09:44
Java
小白也想要offer:简历别放洋屁,搞不还还放错了,当然你投外企除外,以上纯属个人观点
点赞 评论 收藏
分享
Noob1024:一笔传三代,人走笔还在
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
11-24 20:55
阿里国际 Java工程师 2.7k*16.0
程序员猪皮:没有超过3k的,不太好选。春招再看看
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务