• [Codeforces722E] Research Rover (dp+组合数学)


    [Codeforces722E] Research Rover (dp+组合数学)

    题面

    给出一个N*M的方格阵,从(1,1)出发,到(N,M)结束,从(x,y)只能走到(x+1,y)或(x,y+1)。方格阵上还有K个特殊点,初始时给出的分数t每经过一个特殊点就会变成([frac{t}{2}])。求到(N,M)时得分的期望。保证(1,1)和(N,M)不是特殊点。(N,M≤100000, K≤2000, t≤1000000)

    分析

    考虑根据特殊点dp.把(1,1)和(n,m)也看作特殊点(但分数不除2),把特殊点按x排序。(dp[i][j])表示经过至少j个特殊点(包括i),到达i的方案数,g(i,j)表示从i点到j点的方案数,

    [g(i,j)=C_{|x_i-x_j|+|y_i-y_j|}^{|x_i-x_j|} ]

    我们从(1,1)走到(n,m) 向上走n次,向右走m次,一共(n+m)次,从n+m次中选出n次向上走,就是方案数

    [dp[i][j]=sum _{ u,\ x_uleq x_i, y_u leq y_i} (dp[u][j-1] - dp[u][j]) imes g(u,j) ]

    ((dp[u][j-1] - dp[u][j]))表示经过j-1个特殊点到u的方案数。

    为什么我们定义子状态时用的是至少呢?因为从u到i的路径上可能有其他特殊点,导致在u时经过j-1个点,在i时可能就经过j+x (x>=0)个点了,定义为“至少”可以较好的包含这种情况,而需要“正好j个点”的时候只要相减即可。如果我们一开始把子状态定位经过j个点到i的方案数,需要排除中间经过特殊点的情况,容斥起来比较麻烦

    最终答案为:

    $(C_{n+m}^{m} )^{-1} imes sum (dp[k][j]-dp[k][j+1]) imes frac{s}{2^j} $

    代码

    #include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<cmath>
    #include<algorithm>
    #define maxk 3000
    #define maxn 100000
    #define maxlogs 32
    #define mod 1000000007
    using namespace std;
    inline void qread(int &x){
    	x=0;
    	int sign=1;
    	char c=getchar();
    	while(c<'0'||c>'9'){
    		if(c=='-') sign=-1;
    		c=getchar();
    	}
    	while(c>='0'&&c<='9'){
    		x=x*10+c-'0';
    		c=getchar();
    	}
    	x=x*sign;
    }
    typedef long long ll;
    int n,m,k,s;
    int log2s;
    
    struct point{
    	int x;
    	int y;
    	point(){
    		
    	}
    	point(int _x,int _y){
    		x=_x;
    		y=_y;
    	}
    	friend bool operator < (point p,point q){
    		if(p.x==q.x) return p.y<q.y;
    		else return p.x<q.x;
    	} 
    }a[maxk+5];
    
    inline ll fast_pow(ll x,ll k){
    	ll ans=1;
    	while(k){
    		if(k&1) ans=ans*x%mod;
    		x=x*x%mod;
    		k>>=1;
    	}
    	return ans;
    }
    inline ll inv(ll x){
    	return fast_pow(x,mod-2);
    }
    ll fact[maxn*2+5],invfact[maxn*2+5];
    inline ll C(int n,int m){
    //	if(m==0||m==n) return 1;
    //	if(n<m) return 0;
    //不要写这几个判断,常数很大 
    	return fact[n]*invfact[n-m]%mod*invfact[m]%mod;
    }
    void ini_fact(int n){
    	fact[0]=1;
    	for(int i=1;i<=n;i++) fact[i]=fact[i-1]*i%mod;
    	invfact[n]=inv(fact[n]);
    	for(int i=n-1;i>=0;i--) invfact[i]=invfact[i+1]*(i+1)%mod;
    }
    
    
    inline ll calc(point p,point q){
    	return C(q.x-p.x+q.y-p.y,q.x-p.x);
    }
    
    ll dp[maxk+5][maxlogs+5];//dp[i][j]表示当前到第i个点,经过j个点 
    int main(){
    	qread(n);
    	qread(m);
    	qread(k);
    	qread(s);
    	ini_fact(n+m);
    	for(int i=1;i<=k;i++){
    		qread(a[i].x);
    		qread(a[i].y);
    	}
    	sort(a+1,a+1+k);
    	//加入虚拟点(1,1)(n,m) 
    	if(a[1].x!=1||a[1].y!=1){
    		s*=2;//多一个点,会多除一次2,所以把s*2来抵消
    		a[++k]=point(1,1);
    	}
    	if(a[k].x!=n||a[k].y!=m) a[++k]=point(n,m); 
    	else s-=s/2;
    	log2s=log2(s)+1;
    	sort(a+1,a+1+k);
    	dp[1][0]=1;
    	for(int i=2;i<=k;i++){
    		dp[i][1]=calc(a[1],a[i]);
    		for(int j=2;j<=log2s;j++){
    			for(int u=1;u<i;u++){
    				if(a[u].y<=a[i].y&&a[u].x<=a[i].x){
    					dp[i][j]+=dp[u][j-1]*calc(a[u],a[i])%mod;
    					dp[i][j]=(dp[i][j]+mod)%mod;
    					dp[i][j]-=dp[u][j]*calc(a[u],a[i])%mod;
    					dp[i][j]=(dp[i][j]+mod)%mod;
    				}
    			}
    		}
    	} 
    	ll ans=0;
    	for(int i=1;i<=log2s;i++){
    		s-=s/2;
    		ans=(ans+(dp[k][i]-dp[k][i+1]+mod)*s%mod)%mod;
    	}
    	ans=ans*inv(calc(a[1],a[k]))%mod;
    	printf("%I64d
    ",ans);
    } 
    
  • 相关阅读:
    牛客挑战赛45 D.坐标
    树上启发式合并(dsu on tree)合集
    2020HDU多校第二场 1012.String Distance
    2020HDU多校第一场 1009.Leading Robots
    2020牛客暑期多校训练营(第一场)H.Minimum-cost Flow
    自用综合线段树模板(区间加乘、区间置数、区间求和)
    ZOJ 4008.Yet Another Tree Query Problem(问题模型转化+线段树离线处理)
    最小费用最大流模板
    2020 CCPC Wannafly Winter Camp Day3.C. 无向图定向(k染色问题)
    2020牛客寒假算法基础集训营3.E.牛牛的随机数(数位dp拆位算贡献)
  • 原文地址:https://www.cnblogs.com/birchtree/p/11240090.html
Copyright © 2020-2023  润新知