蓝桥杯-蓝跳跳(矩阵快速幂 70%数据)

蓝桥杯-蓝跳跳(矩阵快速幂 70%数据)

题目描述

小蓝制作了一个机器人,取名为蓝跳跳,因为这个机器人走路的时候基本靠跳跃。

蓝跳跳可以跳着走,也可以掉头。蓝跳跳每步跳的距离都必须是整数,每步可以跳不超过 k 的长度。由于蓝跳跳的平衡性设计得不太好,如果连续两次都是跳跃,而且两次跳跃的距离都至少是 p,则蓝跳跳会摔倒,这是小蓝不愿意看到的。

小蓝接到一个特别的任务,要在一个长为 L 舞台上展示蓝跳跳。小蓝要控制蓝跳跳从舞台的左边走到右边,然后掉头,然后从右边走到左边,然后掉头,然后再从左边走到右边,然后掉头,再从右边走到左边,然后掉头,如此往复。

为了让观者不至于太无趣,小蓝决定让蓝跳跳每次用不同的方式来走。小蓝将蓝跳跳每一步跳的距离记录下来,按顺序排成一列,显然这一列数每个都不超过 k 且和是 L。这样走一趟就会出来一列数。如果两列数的长度不同,或者两列数中存在一个位置数值不同,就认为是不同的方案。

请问蓝跳跳在不摔倒的前提下,有多少种不同的方案从舞台一边走到另一边。

输入描述

输入一行包含三个整数 k, p, L。

其中, 1 ≤ p ≤ k ≤ 1000 , 1 ≤ L ≤ 1 0 18 1 ≤ p ≤ k ≤ 1000,1 ≤ L ≤ 10^{18} 1≤p≤k≤1000,1≤L≤1018 。

输出描述

输出一个整数,表示答案。答案可能很大,请输出答案除以 20201114的余数。

思路

由于L的范围肯定直接dp即使可以根据循环减小空间,但时间必然是会超过的,于是想到矩阵快速幂。
F [ i ] [ 0 ] : 一 共 跳 了 i 个 单 位 的 距 离 , 且 当 前 这 一 跳 小 于 p F [ i ] [ 1 ] : 一 共 跳 了 i 个 单 位 的 距 离 , 且 当 前 这 一 跳 大 于 等 于 p F [ i ] [ 0 ] = F [ j ] [ 0 ] + F [ j ] [ 1 ]     j ∈ [ i − p + 1 , i − 1 ] F [ i ] [ 1 ] = F [ j ] [ 0 ]     j ∈ [ i − k , i − p ] F[i][0]:一共跳了i个单位的距离,且当前这一跳小于p \\ F[i][1]:一共跳了i个单位的距离,且当前这一跳大于等于p\\ F[i][0] = F[j][0] + F[j][1] ~~~ j \in{[i-p+1,i-1]}\\ F[i][1] = F[j][0] ~~~j\in[i-k,i-p] F[i][0]:一共跳了i个单位的距离,且当前这一跳小于pF[i][1]:一共跳了i个单位的距离,且当前这一跳大于等于pF[i][0]=F[j][0]+F[j][1]   j∈[i−p+1,i−1]F[i][1]=F[j][0]   j∈[i−k,i−p]
所以可以列出矩阵,以k= 5 ,p = 3 为例:
∣ F [ i ] [ 1 ] F [ i ] [ 0 ] F [ i − 1 ] [ 1 ] F [ i − 1 ] [ 0 ] F [ i − 2 ] [ 1 ] F [ i − 2 ] [ 0 ] F [ i − 3 ] [ 1 ] F [ i − 3 ] [ 0 ] F [ i − 4 ] [ 1 ] F [ i − 4 ] [ 0 ] ∣ = ∣ 0 0 0 0 0 1 0 1 0 1 1 1 1 1 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 0 0 ∣ ∣ F [ i − 1 ] [ 1 ] F [ i − 1 ] [ 0 ] F [ i − 2 ] [ 1 ] F [ i − 2 ] [ 0 ] F [ i − 3 ] [ 1 ] F [ i − 3 ] [ 0 ] F [ i − 4 ] [ 1 ] F [ i − 4 ] [ 0 ] F [ i − 5 ] [ 1 ] F [ i − 5 ] [ 0 ] ∣ \begin{vmatrix} F[i][1]\\ F[i][0]\\ F[i-1][1]\\ F[i-1][0]\\ F[i-2][1]\\ F[i-2][0]\\ F[i-3][1]\\ F[i-3][0]\\ F[i-4][1]\\ F[i-4][0]\\ \end{vmatrix}= \begin{vmatrix} 0 & 0 & 0 & 0 & 0 &1 & 0 & 1 & 0 & 1\\ 1 & 1 & 1 & 1 & 0 &0 & 0 & 0 & 0 & 0\\ 1 & 0 & 0 & 0 & 0 &0 & 0 & 0 & 0 & 0\\ 0 & 1 & 0 & 0 & 0 &0 & 0 & 0 & 0 & 0\\ 0 & 0 & 1 & 0 & 0 &0 & 0 & 0 & 0 & 0\\ 0 & 0 & 0 & 1 & 0 &0 & 0 & 0 & 0 & 0\\ 0 & 0 & 0 & 0 & 1 &0 & 0 & 0 & 0 & 0\\ 0 & 0 & 0 & 0 & 0 &1 & 0 & 0 & 0 & 0\\ 0 & 0 & 0 & 0 & 0 &0 & 1 & 0 & 0 & 0\\ 0 & 0 & 0 & 0 & 0 &0 & 0 & 1 & 0 & 0\\ \end{vmatrix} \begin{vmatrix} F[i-1][1]\\ F[i-1][0]\\ F[i-2][1]\\ F[i-2][0]\\ F[i-3][1]\\ F[i-3][0]\\ F[i-4][1]\\ F[i-4][0]\\ F[i-5][1]\\ F[i-5][0]\\ \end{vmatrix} ∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣​F[i][1]F[i][0]F[i−1][1]F[i−1][0]F[i−2][1]F[i−2][0]F[i−3][1]F[i−3][0]F[i−4][1]F[i−4][0]​∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣​=∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣​0110000000​0101000000​0100100000​0100010000​0000001000​1000000100​0000000010​1000000001​0000000000​1000000000​∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣​∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣​F[i−1][1]F[i−1][0]F[i−2][1]F[i−2][0]F[i−3][1]F[i−3][0]F[i−4][1]F[i−4][0]F[i−5][1]F[i−5][0]​∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣∣​

代码(70%数据)

#include <iostream>
#include <cstring>
using namespace std;
const int N = 2e3 ;
typedef long long ll;
const ll mod = 20201114;
ll f[1005][2];
ll p,k,l;
ll A[N];
ll B[N][N];
ll tmp1[N],tmp2[N][N];
void multi(ll B[N][N], ll A[N] , ll C[N])
{
  memset(tmp1,0,sizeof(tmp1));
  for(int i = 0 ; i < k << 1 ; i ++)
  {
    for(int j = 0 ; j < k << 1; j ++)
    {
      tmp1[i] = (tmp1[i] + B[i][j] * A[j] % mod ) % mod;
    }
  }
  memcpy(C,tmp1,sizeof(tmp1));
}

void multi(ll A[N][N] , ll B[N][N] , ll C[N][N])
{
  memset(tmp2,0,sizeof(tmp2));
  for(int i = 0 ; i < k << 1 ; i ++)
    for(int j = 0 ; j < k << 1 ; j ++)
      for(int m = 0 ; m < k << 1 ; m ++)
        tmp2[i][j] = (tmp2[i][j] + A[i][m] * B[m][j] % mod) % mod;
  memcpy(C,tmp2,sizeof(tmp2));
}

void print(ll A[N])
{
  for(int i = 0 ; i < k << 1 ; i ++)
    printf("%lld " , A[i]);
  puts("");
}

void print(ll A[N][N])
{
  for(int i = 0 ; i < k << 1 ; i ++)
  {
    for(int j = 0 ; j < k << 1 ; j ++)
      printf("%lld ",A[i][j]);
    puts("");
  }
}

int main()
{
  cin>>k>>p>>l;
  f[0][0] = 1;
  for(int i = 1 ; i <= k ; i ++)
  {
    for(int j = 1 ; j < p  && j <= i; j ++)
    {
      f[i][0] = (f[i][0] + (f[i-j][0] + f[i-j][1]) % mod ) % mod;
    }
    for(int j = p ; j <= k && j <= i; j ++)
    {
      f[i][1] = (f[i][1] + f[i-j][0]) % mod;
    }
  }
  if(l <= k)
  {
    cout<<(f[l][0] + f[l][1])%mod;
    return 0;
  }
  for(int i = 0 ; i < k ; i ++)
  {
    A[i<<1]   = f[k-i][1]; 
    A[i<<1|1] = f[k-i][0];
  }
  for(int i = (k<<1) -1 ; i >= (p<<1) - 1 ; i -= 2)
    B[0][i] = 1;
  for(int i = 0 ; i < (p - 1)<<1 ; i ++)
    B[1][i] = 1;
  for(int i = 2 ; i < k << 1 ; i ++)
    B[i][i-2] = 1;
  l -= k;
  while(l)
  {
    if(l & 1)
      multi(B,A,A);
    l >>= 1;
    multi(B,B,B);
  }
  printf("%lld\n",(A[0] + A[1]) % mod);
  return 0;
}
上一篇:[NOI2018] 屠龙勇士


下一篇:算法-链表:链表的常见六个操作