码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj 2159 Crash 的文明世界 —— 第二类斯特林数+树形DP

时间:2018-12-05 12:35:44      阅读:213      评论:0      收藏:0      [点我收藏+]

标签:targe   lan   使用   mem   过程   href   ret   btree   turn   

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2159

使用公式:\( n^{k} = \sum\limits_{i=0}^{k} S(k,i) * i! * C_{n}^{i} \)

所以维护 \( f[x][i] = \sum\limits_{u\in subtree[x],d=dist(x,u)}^{n} C_{d}^{i} \)

然后利用 \( C_{n}^{m} = C_{n-1}^{m} + C_{n-1}^{m-1} \),可以树形DP过程中维护(原来想着可以线段树维护矩阵来着,呵呵)。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=50005,xm=155,mod=10007;
int n,m,hd[xn],ct,to[xn<<1],nxt[xn<<1],f[xn][xm],g[xm],t[xn][xm];
int ans[xn],s[xm][xm],jc[xm];
int rd()
{
  int ret=0,f=1; char ch=getchar();
  while(ch<0||ch>9){if(ch==-)f=0; ch=getchar();}
  while(ch>=0&&ch<=9)ret=ret*10+ch-0,ch=getchar();
  return f?ret:-ret;
}
void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}
int upt(int x){while(x>=mod)x-=mod; while(x<0)x+=mod; return x;}
void dfs(int x,int fa)
{
  f[x][0]=1;
  for(int i=hd[x],u;i;i=nxt[i])
    {
      if((u=to[i])==fa)continue;
      dfs(u,x);
      f[x][0]=upt(f[x][0]+f[u][0]);
      for(int k=1;k<=m;k++)f[x][k]=upt(f[x][k]+f[u][k]+f[u][k-1]);
    }
}
void dfsx(int x,int fa)
{
  if(!fa)memcpy(t[x],f[x],sizeof f[x]);
  else 
    {
      t[x][0]=upt(f[x][0]+g[0]);
      for(int k=1;k<=m;k++)t[x][k]=upt(f[x][k]+g[k]+g[k-1]);//sum of C(d,k)
    }
  for(int i=0;i<=m;i++)
      ans[x]=(ans[x]+(ll)s[m][i]*jc[i]%mod*t[x][i])%mod;
  for(int i=hd[x],u;i;i=nxt[i])
    {
      if((u=to[i])==fa)continue;
      g[0]=upt(t[x][0]-f[u][0]);
      for(int k=1;k<=m;k++)g[k]=upt(t[x][k]-(f[u][k]+f[u][k-1]));
      dfsx(u,x);
    }
}
void init()
{
  jc[0]=1;
  for(int i=1;i<=m;i++)jc[i]=(ll)jc[i-1]*i%mod;
  s[0][0]=1;
  for(int i=1;i<=m;i++)
    for(int j=1;j<=m;j++)
      s[i][j]=upt(s[i-1][j-1]+j*s[i-1][j]);
}
int main()
{
  n=rd(); m=rd(); int L=rd();
  int now=rd(),A=rd(),B=rd(),Q=rd();
  for(int i=1;i<n;i++)
    {
      now = (now * A + B) % Q; 
      int tmp = (i < L) ? i : L; 
      int x=i-now%tmp,y=i+1;
      add(x,y); add(y,x);
    }
  init();
  dfs(1,0); dfsx(1,0);
  for(int i=1;i<=n;i++)printf("%d\n",ans[i]);
  return 0;
}

 

bzoj 2159 Crash 的文明世界 —— 第二类斯特林数+树形DP

标签:targe   lan   使用   mem   过程   href   ret   btree   turn   

原文地址:https://www.cnblogs.com/Zinn/p/10069317.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!