题目大意
一开始有个人在
(x0,y0)
,有
t
个时间点,每个时间点这个人会上下左右四个方向随机选一个方向走。给定
n,m≤100,t≤109
题解
将点的坐标变成
(a+b2,a−b2)
的形式,那么每一步,相当于变成
a
可以
E[(a+b2)n×(a−b2)m]=∑i=0n∑j=0m(ni)(mj)E[(a2)i+j×(b2)n−i×(−b2)m−j]=∑i=0n∑j=0m(ni)(mj)E[(a2)i+j]×E[(b2)n−i]×E[(−b2)m−j]
剩下来的问题就是怎么求 E[ak] ,其中 a 表示最后的坐标对应的
设 Tj,k 表示当前走了 j 步,
Tj,k=∑kp=0(kp)Tj−1,p+∑kp=0(kp)Tj−1,p(−1)k−p2
Tj,k=∑p=0,(k−p)mod2=0k(kp)Tj−1,p
这个由于 n+m≤200 ,因此可以直接矩阵乘法搞。
那么整道题的复杂度就是 O((n+m)3logt) 。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 205,mo = int(1e9) + 7;
typedef int matrix[maxn][maxn];
class RandomWalkOnGrid
{
public:
matrix trans,mat_a,mat_b;
int c[maxn][maxn],__n;
int pow(int a,int b)
{
int tmp = 1;
for(;b;b >>= 1,a = a * 1ll * a % mo)
if (b & 1) tmp = tmp * 1ll * a % mo;
return tmp;
}
void mul(matrix &a,matrix &b,matrix &c)
{
static matrix t;
memset(t,0,sizeof t);
for(int i = 0;i <= __n;i ++)
for(int k = 0;k <= __n;k ++)
if (a[i][k])
for(int j = 0;j <= __n;j ++)
if (b[k][j])
t[i][j] = (t[i][j] + a[i][k] * 1ll * b[k][j]) % mo;
memcpy(c,t,sizeof c);
}
void pow(matrix &a,int b)
{
matrix tmp;
memset(tmp,0,sizeof tmp);
for(int i = 0;i <= __n;i ++) tmp[i][i] = 1;
for(;b;b >>= 1,mul(a,a,a))
if (b & 1) mul(tmp,a,tmp);
memcpy(a,tmp,sizeof a);
}
int rev(int a)
{
return pow(a,mo - 2);
}
int getExpectation(int x0,int y0,int t,int n,int m)
{
__n = n + m;
for(int i = 1;i <= __n;i ++) c[i][0] = c[0][i] = 1;
for(int i = 1;i <= __n;i ++)
for(int j = 1;j <= i;j ++)
c[i][j] = (c[i - 1][j - 1] + c[i - 1][j]) % mo;
c[0][0] = 1;
for(int k = 0;k <= __n;k ++)
for(int p = 0;p <= k;p ++)
if ((k - p) % 2 == 0) trans[p][k] = 2ll * c[k][p] % mo;
pow(trans,t);
int ans = 0,a = ((x0 + y0) % mo + mo) % mo,b = ((x0 - y0) % mo + mo) % mo;
for(int i = 0;i <= __n;i ++) mat_a[0][i] = pow(a,i),mat_b[0][i] = pow(b,i);
mul(mat_a,trans,mat_a),mul(mat_b,trans,mat_b);
for(int i = 0;i <= n;i ++)
for(int j = 0;j <= m;j ++)
{
int cur = mat_a[0][i + j] * 1ll * mat_b[0][n - i + m - j] % mo;
if ((m - j) & 1) cur = (mo - cur) % mo;
cur = cur * 1ll * rev(pow(2,n + m)) % mo;
ans = (ans + cur * 1ll * c[n][i] % mo * c[m][j] % mo) % mo;
}
return ans;
}
};