• LOJ#3124. 「CTS2019 | CTSC2019」氪金手游 容斥+DP


    神仙容斥+DP可还行.  

    code: 

    #include <cstdio> 
    #include <cmath> 
    #include <vector>
    #include <cstring>
    #include <algorithm>     
    #define N 1007  
    #define ll long long   
    #define mod 998244353 
    #define pb push_back 
    #define setIO(s) freopen(s".in","r",stdin)  
    using namespace std;
    int n,a[N][4],size[N<<2],f[N][N<<2],tmp[N<<2];    
    struct Edge {   
    	int ty,v;   
    	Edge(int ty=0,int v=0):ty(ty),v(v){}  
    };  
    vector<Edge>G[N];  
    int qpow(int x,int y) { 
    	int tmp=1; 
    	for(;y;y>>=1,x=(ll)x*x%mod) { 
    		if(y&1) tmp=(ll)tmp*x%mod;  
    	} 
    	return tmp; 
    }  
    inline int get_inv(int x) { 
    	return qpow(x,mod-2); 
    }  
    inline int ADD(int x,int y) { 
    	return (x+y)>=mod?x+y-mod:x+y;  
    }  
    inline int DEC(int x,int y) { 
    	return (x-y)<0?(x-y+mod):x-y;   
    }
    void dfs(int x,int ff) {  
    	size[x]=3;   
    	f[x][1]=a[x][1],f[x][2]=a[x][2],f[x][3]=a[x][3];   
    	for(int i=0;i<G[x].size();++i) { 	
    		Edge e=G[x][i];  
    		if(e.v==ff) continue;   
    		dfs(e.v,x);   	  
    		int y=e.v; 
    		for(int j=1;j<=size[x]+size[y];++j) {	 	
    			tmp[j]=0; 
    		}
    		if(e.ty==0) {
    			for(int j=1;j<=size[x];++j) { 
    				for(int k=1;k<=size[y];++k) {  
    					int cur=(ll)f[x][j]*f[y][k]%mod;   
    					tmp[j+k]=DEC(tmp[j+k],cur);   
    					tmp[j]=ADD(tmp[j],cur);        
    				}
    			}
    		} 
    		else {  	
    			for(int j=1;j<=size[x];++j) { 	
    				for(int k=1;k<=size[y];++k) { 	
    					tmp[j+k]=ADD(tmp[j+k],(ll)f[x][j]*f[y][k]%mod);  
    				}
    			}
    		}
    		size[x]+=size[y];  
    		for(int j=1;j<=size[x];++j) f[x][j]=tmp[j];  
    	}   
    	for(int i=1;i<=size[x];++i) f[x][i]=(ll)f[x][i]*get_inv(i)%mod;  
    }
    int main() { 
    	// setIO("input"); 
    	scanf("%d",&n);   
    	int x,y,z; 
    	for(int i=1;i<=n;++i) { 
    		scanf("%d%d%d",&x,&y,&z);  
    		ll s=x+y+z;  
    		s=get_inv(s%(mod-2));     
    		a[i][1]=(ll)x*s%mod;  
    		a[i][2]=(ll)2*y%mod*s%mod;  
    		a[i][3]=(ll)3*z%mod*s%mod;  
    	}     
    	for(int i=1;i<n;++i) { 	
    		scanf("%d%d",&x,&y); 
    		G[x].pb(Edge(1,y));  
    		G[y].pb(Edge(0,x)); 
    	}  
    	dfs(1,0);  
    	int ans=0; 
    	for(int i=1;i<=size[1];++i) {  
    		ans=ADD(ans,f[1][i]);  
    	}  
    	printf("%d
    ",ans);  
    	return 0;
    }
    

      

  • 相关阅读:
    Oracle--pl/sql编程-分支语句(判断、循环)
    axios二次封装及API接口统一管理
    Vuejs之axios获取Http响应头
    对webpack和gulp的理解和区别
    nodejs+redis使用
    Linux安装Nginx、Redis、django
    js 原型里面写方法
    在layui中使用 jquery 触发select 的 change事件无效
    promise看这篇就够了
    vue 封装原型方法 加 promist .then
  • 原文地址:https://www.cnblogs.com/guangheli/p/13433264.html
Copyright © 2020-2023  润新知