CodeCraft-21 and Codeforces Round #711 (Div. 2) C. Planar Reflections
题意
假设现在有 n n n个平面,从左到右排列
现有一个衰变年龄为 k k k的粒子从最左端向右发射
如果一个粒子碰到了一个平面,它将会不改变衰变年龄保持运动方向并传过这个平面
如果这个粒子的衰变年龄大于 1 1 1,那么将会分裂出一个衰变年龄为 k − 1 k-1 k−1的往相反方向运动的粒子
给定 n n n和 k k k,问这个空间最终会有多少个粒子,答案对 1 0 9 + 7 10^9+7 109+7取模
限制
1 ≤ T ≤ 100 1 ≤ n , k ≤ 1000 ∑ n ≤ 1000 ∑ k ≤ 1000 1\le T\le 100\\ 1\le n,k\le 1000\\ \sum n\le 1000\\ \sum k\le 1000 1≤T≤1001≤n,k≤1000∑n≤1000∑k≤1000
思路
令 d p [ i ] [ j ] dp[i][j] dp[i][j]表示某个粒子运动方向的前方还有 i i i个平面未穿过,且**衰变年龄为 j j j**时,这个粒子所能分裂出的粒子数量
那么答案明显就是 d p [ n ] [ k ] dp[n][k] dp[n][k]
首先,如果某个粒子的衰变年龄为 1 1 1,那么不管前面有多少平面,贡献恒为 1 1 1,即 d p [ x ] [ 1 ] = 1 dp[x][1]=1 dp[x][1]=1
然后,如果某个粒子前面没有平面,也就无法继续分裂,贡献恒为 1 1 1,即 d p [ 0 ] [ x ] = 1 dp[0][x]=1 dp[0][x]=1
最后考虑分裂的状态转移:
原本粒子向前走,衰变年龄不变,即 d p [ i − 1 ] [ j ] dp[i-1][j] dp[i−1][j]
分裂出的粒子向后走,衰变年龄减一,即 d p [ n − i ] [ j − 1 ] dp[n-i][j-1] dp[n−i][j−1]
故得到状态转移方程 d p [ i ] [ j ] = d p [ i − 1 ] [ j ] + d p [ n − i ] [ j − 1 ] dp[i][j]=dp[i-1][j]+dp[n-i][j-1] dp[i][j]=dp[i−1][j]+dp[n−i][j−1]
考虑到引用关系,需要先枚举 j j j再枚举 i i i进行处理
代码
//#include<ext/pb_ds/assoc_container.hpp>
//#include<ext/pb_ds/hash_policy.hpp>
#include<bits/stdc++.h>
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)
#define pb push_back
#define eb emplace_back
#define mst(a,b) memset(a,b,sizeof(a))
using namespace std;
//using namespace __gnu_pbds;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> P;
const int INF=0x3f3f3f3f;
const ll LINF=0x3f3f3f3f3f3f3f3f;
const double eps=1e-12;
const double PI=acos(-1.0);
const double angcst=PI/180.0;
const ll mod=1e9+7;
ll gcd(ll a,ll b){return b==0?a:gcd(b,a%b);}
ll qmul(ll a,ll b){ll r=0;while(b){if(b&1)r=(r+a)%mod;b>>=1;a=(a+a)%mod;}return r;}
ll qpow(ll a,ll n){ll r=1;while(n){if(n&1)r=(r*a)%mod;n>>=1;a=(a*a)%mod;}return r;}
ll qpow(ll a,ll n,ll p){ll r=1;while(n){if(n&1)r=(r*a)%p;n>>=1;a=(a*a)%p;}return r;}
int dp[1050][1050];
void solve()
{
int n,k;
cin>>n>>k;
rep(i,0,n)
rep(j,0,k)
dp[i][j]=0;
rep(i,1,n)
dp[i][1]=1;
rep(i,1,k)
dp[0][i]=1;
rep(j,1,k)
rep(i,1,n)
dp[i][j]=(dp[i-1][j]+dp[n-i][j-1])%mod;
cout<<dp[n][k]<<'\n';
}
int main()
{
closeSync;
multiCase
{
solve();
}
return 0;
}