首页 > 代码库 > 2432: [Noi2011]兔农 - BZOJ

2432: [Noi2011]兔农 - BZOJ

Description

 农夫栋栋近年收入不景气,正在他发愁如何能多赚点钱时,他听到隔壁的小朋友在讨论兔子繁殖的问题。

问题是这样的:第一个月初有一对刚出生的小兔子,经过两个月长大后,这对兔子从第三个月开始,每个月初生一对小兔子。新出生的小兔子生长两个月后又能每个月生出一对小兔子。问第n个月有多少只兔子?

聪明的你可能已经发现,第n个月的兔子数正好是第n个Fibonacci(斐波那契)数。栋栋不懂什么是Fibonacci数,但他也发现了规律:第i+2个月的兔子数等于第i个月的兔子数加上第i+1个月的兔子数。前几个月的兔子数依次为:

1 1 2 3 5 8 13 21 34 …

栋栋发现越到后面兔子数增长的越快,期待养兔子一定能赚大钱,于是栋栋在第一个月初买了一对小兔子开始饲养。

每天,栋栋都要给兔子们喂食,兔子们吃食时非常特别,总是每k对兔子围成一圈,最后剩下的不足k对的围成一圈,由于兔子特别害怕孤独,从第三个月开始,如果吃食时围成某一个圈的只有一对兔子,这对兔子就会很快死掉。

我们假设死去的总是刚出生的兔子,那么每个月的兔子数仍然是可以计算的。例如,当k=7时,前几个月的兔子数依次为:

1 1 2 3 5 7 12 19 31 49 80 …

给定n,你能帮助栋栋计算第n个月他有多少对兔子么?由于答案可能非常大,你只需要告诉栋栋第n个月的兔子对数除p的余数即可。
Input



输入一行,包含三个正整数n, k, p。
Output

 

输出一行,包含一个整数,表示栋栋第n个月的兔子对数除p的余数。
Sample Input


6 7 100

Sample Output


7
HINT

 1<=N<=10^18

2<=K<=10^6

2<=P<=10^9

 

 

简单题解:斐波那契数列在mod k的意义下有循环节,长度在6*k之内

我们要求的数列mod k之后是这样的1,1,2,3....a,0,1*a,1*a,2*a .... b,0,1*b 2*b ... 0

1 a b...这个数列也是有循环节的(可以通过求逆元求出这个数列),这个循环节长度在k以内(可能没有循环,后面mod k都不为1),然后加上矩阵快速幂即可

详细题解:斐波那契数列在mod k的意义下有循环节,长度在6*k之内

因为*/%……@%@¥%@¥#%…*&!@¥@¥%&*/@%

你们还是看VFK的吧

  1 const  2     maxk=1001000;  3 type  4     matrix=array[1..3,1..3]of int64;  5 var  6     a:array[0..maxk*6]of longint;  7     first,last,b,len:array[0..maxk]of longint;  8     pre:array[0..maxk]of longint;  9     n,k,p,x,y:int64; 10     ans,aa,bb:matrix; 11  12 procedure exgcd(a,b:int64); 13 var 14     t:int64; 15 begin 16     if b=0 then 17     begin 18         if a=1 then x:=1 19         else x:=0; 20         y:=0; 21         exit; 22     end; 23     exgcd(b,a mod b); 24     t:=x; 25     x:=y; 26     y:=t-(a div b)*y; 27 end; 28  29 procedure work; 30 var 31     i:longint; 32 begin 33     a[1]:=1;a[2]:=1;i:=2; 34     while true do 35         begin 36             inc(i); 37             a[i]:=(a[i-1]+a[i-2])mod k; 38             if first[a[i]]=0 then first[a[i]]:=i; 39             if (a[i]=1) and (a[i-1]=1) then break; 40         end; 41     for i:=1 to k-1 do 42         begin 43             exgcd(i,k); 44             pre[i]:=(x mod k+k)mod k; 45         end; 46     b[1]:=1;last[1]:=1;i:=1; 47     while true do 48         begin 49             len[i]:=first[pre[b[i]]]-1; 50             if len[i]<0 then break; 51             inc(i); 52             b[i]:=int64(a[len[i-1]])*b[i-1]mod k; 53             if last[b[i]]>0 then break; 54             last[b[i]]:=i; 55         end; 56     aa[1,2]:=1;aa[2,1]:=1;aa[2,2]:=1;aa[3,3]:=1; 57     bb:=aa;bb[3,2]:=p-1; 58 end; 59  60 operator *(a,b:matrix)c:matrix; 61 var 62     i,j,k:longint; 63 begin 64     fillchar(c,sizeof(c),0); 65     for i:=1 to 3 do 66         for j:=1 to 3 do 67             for k:=1 to 3 do 68                 c[i,k]:=(c[i,k]+a[i,j]*b[j,k])mod p; 69 end; 70  71 function f(a:matrix;n:int64):matrix; 72 begin 73     fillchar(f,sizeof(f),0); 74     f[1,1]:=1;f[2,2]:=1;f[3,3]:=1; 75     while n>0 do 76         begin 77             if n and 1=1 then f:=f*a; 78             a:=a*a; 79             n:=n>>1; 80         end; 81 end; 82  83 procedure main; 84 var 85     i,j:longint; 86     s:matrix; 87     sum:int64; 88 begin 89     read(n,k,p); 90     if n<3 then 91     begin 92         writeln(1); 93         exit; 94     end; 95     work; 96     ans[1,1]:=1;ans[2,2]:=1;ans[3,3]:=1; 97     if n>len[1] then 98         begin 99             dec(n,len[1]+1);100             ans:=ans*f(aa,len[1]-2)*bb;101         end102     else103         begin104             ans:=ans*f(aa,n-2);105             n:=0;106         end;107     i:=2;108     while n>0 do109         begin110             if pre[b[i]]=0 then111             begin112                 ans:=ans*f(aa,n);113                 n:=0;114                 break;115             end;116             if len[i]<0 then117             begin118                 ans:=ans*f(aa,n);119                 n:=0;120                 break;121             end;122             if last[b[i]]<i then break;123             if n>len[i] then124                 begin125                     dec(n,len[i]+1);126                     ans:=ans*f(aa,len[i])*bb;127                 end128             else129                 begin130                     ans:=ans*f(aa,n);131                     n:=0;132                 end;133             inc(i);134         end;135     if n=0 then136     begin137         writeln((ans[1,2]+ans[2,2]+ans[3,2])mod p);138         exit;139     end;140     j:=i;141     sum:=0;142     fillchar(s,sizeof(s),0);143     s[1,1]:=1;s[2,2]:=1;s[3,3]:=1;144     for i:=last[b[j]] to j-1 do145         begin146             inc(sum,len[i]+1);147             s:=s*f(aa,len[i])*bb;148         end;149     ans:=ans*f(s,n div sum);150     n:=n mod sum;151     i:=last[b[j]];152     while n>0 do153         begin154             if n>len[i] then155                 begin156                     ans:=ans*f(aa,len[i])*bb;157                     dec(n,len[i]+1);158                 end159             else160                 begin161                     ans:=ans*f(aa,n);162                     n:=0;163                 end;164             inc(i);165         end;166     writeln((ans[1,2]+ans[2,2]+ans[3,2])mod p);167 end;168 169 begin170     main;171 end.
View Code