Jump on Axis (逆元,组合数)

Jump on Axis

题目链接

题意 :
最开始在0，要跳到M。(M <= 1e7) 每次可以选择1，2，3中的一个数。如果选的数之前没选，就跳那么多步，否则第i次选j的话就跳3*(i-1)+j步。输出y，z。y表示最少多少步能跳到M，z表示跳到M的不同跳法总数（mod 1e9+7）。多组输入（T<=200）
思路 :
选1，2，3的次数分别设为i，j，k。可以求出i最大为2583，那么枚举i，j可以直接得到k的个数，每次更新y=min(y,i+j+k)，z即可。

z咋更新。视作i+j+k种选i个数安排1，然后选j个位置安排j。即C(i+j+k,i)*C(j+k,j)即可。
*AC代码 :

#include <bits/stdc++.h>

#define int long long
#define pb push_back
using namespace std;

const int N = 3e3 * 3 + 10; 
const int mod = 1e9 + 7;
const int INF = 1e9;

int n;
vector<int> v[5];

int pow(int a, int p, int mod)
{
    int ans = 1;
    while (p)
    {
        if (p & 1)
            ans = ans * a % mod;
        p >>= 1, a = (a * a) % mod;
    }
    return ans;
}

int inv[N], re[N], fac[N];
void init(int n)
{

    for (int i = 1; i <= 3; i++)
    {
        v[i].pb(0);
        for (int j = 1;; j++)
        {
            int x = v[i][j - 1] + 3 * (j - 1) + i;
            if (x > 1e7)
                break;
            v[i].pb(x);
        }
        v[i].pb(INF);
    }

    inv[0] = re[0] = fac[0] = 1;
    for (int i = 1; i <= n; i++)
    {
        fac[i] = fac[i - 1] * i % mod;
        re[i] = pow(fac[i], mod - 2, mod);
    }
}

signed main()
{
    int T;
    scanf("%lld", &T);
    init(N - 1);

    for (int _ = 1; _ <= T; _++)
    {
        scanf("%lld", &n);
        int mi = INF, sum = 0;
        int i, j, k, x;
        for (i = 0; v[1][i] <= n; i++)
        {
            for (j = 0; v[1][i] + v[2][j] <= n; j++)
            {
                x = n - (v[1][i] + v[2][j]);
                k = lower_bound(v[3].begin(), v[3].end(), x) - v[3].begin();
                if (v[3][k] == x)
                {
                    mi = min(mi, i + j + k);
                    sum = (sum + fac[i + j + k] * re[i] % mod * re[j] % mod * re[k] % mod) % mod;
                }
            }
        }
        printf("Case %lld: %lld %lld\n", _, mi, sum);
    }
    
    return 0;
}