HDU 6050 Funny Function （规律+逆元）

Description

Function $F_{x,y}$ satisfies:

$F_{1,1}=F_{1,2}=1$

$F_{1,i}=F_{1,i-1}+2*F_{1,i-2}\ (i>=3)$

$F_{i,j}=\sum_{k=j}^{j+N-1}F_{i-1,k}\ (i>=2,j>=1)$

For given integers N and M,calculate $F_{m,1}$ modulo 1e9+7.

 

Input

There is one integer T in the first line.

The next T lines,each line includes two integers N and M .

1<=T<=10000,1<=N,M<2^63.

 

Output

For each given N and M,print the answer in a single line.

 

Sample Input

2
2 2
3 3

 

Sample Output

2
33

 

题意

给出 n , m 的值，求 Fm,1 mod 1e9+7 。

 

思路

先看数据范围： $1<=N,M<2^{63}$ ，当然不会允许遍历之类的计算方法咯~

所以最终的结果一定是有规律，并且可以通过公式计算出来的。

先打表看看

// 当 n = 2 时
1 1 3 5 11 21 43 85 171 341
2 4 8 16 32 64 128 256 512 1024
6 12 24 48 96 192 384 768 1536 3072
18 36 72 144 288 576 1152 2304 4608 9216
54 108 216 432 864 1728 3456 6912 13824 27648
162 324 648 1296 2592 5184 10368 20736 41472 82944
486 972 1944 3888 7776 15552 31104 62208 124416 248832
1458 2916 5832 11664 23328 46656 93312 186624 373248 746496
4374 8748 17496 34992 69984 139968 279936 559872 1119744 2239488

// 当 n = 3 时
1 1 3 5 11 21 43 85 171 341
5 9 19 37 75 149 299 597 1195 2389
33 65 131 261 523 1045 2091 4181 8363 16725
229 457 915 1829 3659 7317 14635 29269 58539 117077
1601 3201 6403 12805 25611 51221 102443 204885 409771 819541
11205 22409 44819 89637 179275 358549 717099 1434197 2868395 5736789
78433 156865 313731 627461 1254923 2509845 5019691 10039381 20078763 40157525
549029 1098057 2196115 4392229 8784459 17568917 35137835 70275669 140551339 281102677
3843201 7686401 15372803 30745605 61491211 122982421 245964843 491929685 983859371 1967718741

首先可以明确的是 $F_{1,1}=1$ 。

---

我们来看当 $n$ 为偶数的情况，第一列即 $F_{m,1}$ ，发现它们之间存在倍数 $3$ 的关系（除第二行以外），读者可以自己写程序测试其他数据，最终发现这个倍数刚好是 $2^n-1$ ，于是对于偶数， $F_{m,1}=F_{2,1}×(2^n-1)^{m-2}$ 。

---

然后我们来看当 $n$ 为奇数的情况，还照之前的做法我们把相邻的两项相除，可以发现这个值无限趋近于 $2^n-1$ ，于是想到会不会相差某一个数字呢，然后计算 $F_{m-1,1}×(2^n-1)-F_{m,1}$ 发现果真相差一个常数，接下来的工作便是寻找这个常数与什么有关了。

当 $n=3..5..7..9$ 时，该常数为： $2..10..42..170$ ，有规律么？

答案当然是有啦~

$2=2^1$

$10=2^1+2^3$

$42=2^1+2^3+2^5$

$170=2^1+2^3+2^5+2^7$

也就是公比为 $2^2$ 的等比数列前 $k$ 项和，公式： $S_k=\frac{a_1(1-q^k)}{1-q}$ ，其中 $q=2^2,a_1=2,k=\frac{n}{2}$ 。

 

我们设 $cha$ 为刚刚我们所求得的差， $F_{2,1}$ 已知、 $2^n-1$ 已知，然后我们如何通过这三者得到 $F_{m,1}$ 呢？

$F_{3,1}=F_{2,1}×(2^n-1)-cha$

$F_{4,1}=F_{3,1}×(2^n-1)-cha=F_{2,1}×(2^n-1)^2-cha×(2^n-1)-cha$

$F_{5,1}=F_{4,1}×(2^n-1)-cha=F_{2,1}×(2^n-1)^3-cha×(2^n-1)^2-cha×(2^n-1)-cha$

…

$F_{m,1}=F_{m-1,1}×(2^n-1)-cha=F_{2,1}×(2^n-1)^{m-2}-cha×(2^n-1)^{m-3}...-cha$

显然，后面的 $cha×(2^n-1)^{m-3}+cha×(2^n-1)^{m-2}...+cha×(2^n-1)+cha$ 又是一个公比为 $2^n-1$ 的等比数列前 $k$ 项和，同样我们可以代入公式求得，我们设其为 $add$ 。

于是，最终的结果便为： $F_{m,1}=F_{2,1}×(2^n-1)^{m-2}-add$

---

此时，整个式子中的未知量只有 $F_{2,1}$ 了，那它怎么得到呢？遍历相加？会超时？

同样也有规律的，先打表看看~

S[i]: 1 2 5 10 21 42 85 170 341 682 1365 2730 5461 10922 21845 43690 87381 174762 349525
F[i]: 1 1 3 5 11 21 43 85 171 341 683 1365 2731 5461 10923 21845 43691 87381 174763 349525

$S_i$ 代表 $F_{1,i}$ 的前 $i$ 项和，将 $S$ 统一右移一个会发现什么呢？

当 $i$ 为奇数时 $S_i=F_{i+1}$ ，否则 $S_i=F_{i+1}-1$

因此我们只需要计算 $F_{i+1}$ 便可以得到 $S_i$ ，而 $F_{1,i+1}=\frac{(-1)^i+2^{i+1}}{3}$ （PS: 这里也是需要找规律求得）。

然后整个问题便可以求解啦~

注意：因为是取模运算，因此在计算中一切除法全部用乘法逆元来代替，记得随时取模以防溢出。

 

AC 代码

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
#include<iostream>
using namespace std;

typedef __int64 LL;
const int mod = 1e9+7;
LL n,m;
LL mult(LL a,LL n)
{
    LL ans=1;
    while(n)
    {
        if(n&1)ans=(ans*a)%mod;
        a=(a*a)%mod;
        n>>=1;
    }
    return ans;
}

void solve()
{
    if(m==1)
    {
        cout<<"1"<<endl;
        return;
    }
    LL three=mult(3,mod-2);         // 3 的逆元
    LL one=((mult(2,n+1)+(n&1?-1:1)+mod)%mod*three)%mod;    // F[1][n+1]
    if(n%2==0)one=(one-1+mod)%mod;  // 计算 F[2][1]
    LL ans=0;
    LL mu=(mult(2,n)-1+mod)%mod;    // 2^n-1
    LL mun=mult(mu,m-2);            // (2^n-1)^(m-2)
    if(n%2==0)                      // 偶数时
        ans=(one*mun)%mod;
    else                            // 奇数时
    {
        LL cha=n/2;
        cha=((2*(mult(4,cha)-1+mod)%mod)%mod)*three%mod;
        LL add=((cha*(mun-1+mod)%mod)*mult((mu-1+mod)%mod,mod-2))%mod;
        ans=((one*mun)%mod-add+mod)%mod;
    }
    cout<<ans<<endl;
}

int main()
{
    ios::sync_with_stdio(false);
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n>>m;
        solve();
    }
    return 0;
}