兔子跳跃之谜下

兔子跳跃之谜下

bzoj权限题，但这个网址可免费评测

题目描述
小生和小森在玩兔子之谜游戏。有三只兔子排成一排。知道每只兔子的初始位置，以及三个兔窝的位置。

游戏的规则是，重复以下步骤k次：选择两个不同的兔子A和B，分别位于a和b。A可以跳过B到达2*b-a的点：

跳跃是不允许其他小兔子已经在点2*b-a的位置上：

跳跃也不允许一次跳过一个以上的兔子：

现在小生和小森想要知道，k次操作之后，能否让所有兔子都分别跳到一个兔窝里面。注意，第i个兔子并不一定要在第i个巢。并且输出跳法的种数，数值可能很大，要对结果取模1000000007。只要有一个跳跃是不同的，两种方式被认为是不同的。

输入
有多组测试数据：
第一行，包含一个整数Num，表示测试数据的个数。（1<=Num<=10）
每组测试数据，
第一行三个整数，第二行三个整数，分别表示兔子的初始位置和兔窝的位置。两组数值都是严格递增给出。范围均为[-10^18,10^18]。
最后一个整数k。[1,100]。

输出
共Num行，
跳跃的种数。

样例输入
8
0 5 8
0 8 11
1
0 5 8
0 8 11
3
0 5 8
0 8 11
2
5 8 58
13 22 64
58
0 1 2
1 2 3
100
5 8 58
20 26 61
58
67 281 2348
235 1394 3293
83
-1000000000000000000 999999999999999998 999999999999999999
-1000000000000000000 999999999999999999 1000000000000000000
5
样例输出
1
5
0
0
0
537851168
167142023
29
提示

此题只有三只兔子

可发现规律：向里跳只有一种可能，向外跳有两种可能

由此想到了树，（初始状态）为树上一个节点，要走到（终止状态）的方案总数

lca=（初始状态）与（终止状态）最近公共祖先

f[i][j][k]表示（初始状态）距离(lca)为i，（终止状态）距离（lca）为j，还剩k步的方案总数。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define M 1000000007
#define ll long long 
ll f[105][105][105];
ll d[5],e[5];
ll dp1[105][5],dp2[105][5];
ll _a,_b,_c;
void up(ll a, ll b, ll c){
    if (b-a<c-b)
        _a=b,_b=2*b-a,_c=c;
    else
        _a=a,_b=2*b-c,_c=b;
}
ll dfs(ll i, ll j, ll k, ll a, ll b, ll c){
    //cout<<i<<" "<<j<<" "<<k<<endl;
    if (f[i][j][k]!=-1) return f[i][j][k]%M;


    if ((2*b)==(a+c)){
        if (i>0)
            f[i][j][k]=(dfs(i+1,j,k-1,2*a-b,a,c)+dfs(i+1,j,k-1,a,c,2*c-b))%M;
        if (i==0&&j>0)
            f[i][j][k]=(dfs(i+1,j,k-1,2*a-b,a,c)+dfs(0,j-1,k-1,a,c,2*c-b))%M;
        if (i==0&&j==0)
            f[i][j][k]=(dfs(1,0,k-1,2*a-b,a,c)+dfs(1,0,k-1,a,c,2*c-b))%M;
        return f[i][j][k]%M;
    }
    else{
        if (i>0){
            up(a,b,c);
            f[i][j][k]=(dfs(i-1,j,k-1,_a,_b,_c)+dfs(i+1,j,k-1,2*a-b,a,c)+dfs(i+1,j,k-1,a,c,2*c-b))%M;
        }
        if (i==0&&j>0){
            up(a,b,c);
            f[i][j][k]=(dfs(0,j+1,k-1,_a,_b,_c)+dfs(1,j,k-1,2*a-b,a,c)+dfs(0,j-1,k-1,a,c,2*c-b))%M;
        }
        if (i==0&&j==0){
            up(a,b,c);
            f[i][j][k]=(dfs(0,1,k-1,_a,_b,_c)+dfs(1,0,k-1,2*a-b,a,c)+dfs(1,0,k-1,a,c,2*c-b))%M;
        }
        return f[i][j][k]%M;
    }
}
int main(){
  int cas;
  scanf("%d",&cas);
  while (cas--){

    scanf("%lld%lld%lld",&d[0],&d[1],&d[2]);
    for (int i=0;i<=2;++i)
      dp1[0][i]=d[i];

    scanf("%lld%lld%lld",&e[0],&e[1],&e[2]);
    for (int i=0;i<=2;++i)
      dp2[0][i]=e[i]; 

    int z;
    scanf("%d",&z);  
    for (int i=1;i<=z;++i){
      if (dp1[i-1][1]-dp1[i-1][0]<dp1[i-1][2]-dp1[i-1][1]){
        dp1[i][0]=dp1[i-1][1];
        dp1[i][1]=dp1[i-1][1]*2-dp1[i-1][0];
        dp1[i][2]=dp1[i-1][2];
      }else if (dp1[i-1][1]-dp1[i-1][0]>dp1[i-1][2]-dp1[i-1][1]){
        dp1[i][0]=dp1[i-1][0];
        dp1[i][1]=dp1[i-1][1]*2-dp1[i-1][2];
        dp1[i][2]=dp1[i-1][1];
      }else
        break;
    }
    int ans1=-1,ans2=-1;
    for (int i=0;i<=z;++i){
      for (int j=0;j<=z;++j)
      if (dp1[j][0]==dp2[i][0]&&dp1[j][1]==dp2[i][1]&&dp1[j][2]==dp2[i][2]){
        ans1=i;
        ans2=j;
        break;
      }
      if (ans1!=-1) break;
      if (i==z) break;
      if (dp2[i][1]-dp2[i][0]<dp2[i][2]-dp2[i][1]){
        dp2[i+1][0]=dp2[i][1];
        dp2[i+1][1]=dp2[i][1]*2-dp2[i][0];
        dp2[i+1][2]=dp2[i][2];
      }else if(dp2[i][1]-dp2[i][0]>dp2[i][2]-dp2[i][1]){
        dp2[i+1][0]=dp2[i][0];
        dp2[i+1][1]=dp2[i][1]*2-dp2[i][2];
        dp2[i+1][2]=dp2[i][1];
      }else
        break;
    }
    if (ans1==-1){
      printf("0\n");
      continue;
    }
    for (int i=0;i<=100;++i)
      for (int j=0;j<=100;++j)
        for (int k=0;k<=100;++k)
          f[i][j][k]=-1;
    for (int i=0;i<=100;++i)
      for (int j=0;j<=100;++j) f[i][j][0]=0;
    f[0][0][0]=1;
    printf("%lld\n",dfs(ans1,ans2,z,e[0],e[1],e[2])%M);
  }
  return 0;
}