1. 程式人生 > >bzoj 3027 [Ceoi2004] Sweet —— 生成函式

bzoj 3027 [Ceoi2004] Sweet —— 生成函式

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=3027

就是 (1+x+x2+...+xm[i]) 乘起來;

原來想和揹包一樣做,然而時限很短,陣列也開不了很多,本來以為勉強一下也可以,後來突然發現不行...

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const xn=1e7+5,mod=2004;
int n,a,b,s[xn],m;
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<<3)+(ret<<1)+ch-'0',ch=getchar(); return f?ret:-ret; } int upt(int x){while(x>=mod)x-=mod; while(x<0)x+=mod; return x;} int main() { n
=rd(); a=rd(); b=rd(); m=rd(); for(int i=0;i<=m;i++)s[i]=i+1; for(int i=m+1;i<=b;i++)s[i]=s[m];// for(int i=2;i<=n;i++) { m=rd(); //mx=min(mx+m[i],b); for(int j=b;j>=0;j--)// if(j-m-1>=0)s[j]=upt(s[j]-s[j-m-1]); for(int j=1;j<=b;j++)s[j]=upt(s[j]+s[j-1
]); } int ans=s[b]; if(a)ans=upt(ans-s[a-1]); printf("%d\n",ans); return 0; }
TLE

首先,要化簡這個多項式,得到 ∏(1-xm[i]+1) / (1-x)n

可以把分子和分母分開,分母就是熟悉的 ∑ C(n+i-1,n-1)*xi

而分子一共只有 n 項,可以 2n 搜出每個係數;

然後把二者組合在一起,對於搜出的 k * xy ,對答案有貢獻還需要把 xy 變成 xa ~ xb

所以對應分母多項式的 xa-y ~ xb-y 的係數,是連續的組合數求和,楊輝三角里的一列;

但是模數不是質數,所以組合數不好算;

參考TJ,竟然可以對組合數和模數都乘 n!,就可以 O(n) 直接乘得到組合數了,最後把答案除以 n! 即可;

如果把搜到的係數存下來,最後遍歷,複雜度反而成了 O(bn) ... 不如直接在搜尋裡計算,有值才算上,複雜度 O(n*2n)。

程式碼如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=15,xm=1e7+5,mod=2004;
int n,a,b,m[xn];
ll fac,p,ans;
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<<3)+(ret<<1)+ch-'0',ch=getchar();
  return f?ret:-ret;
}
ll upt(ll x){while(x>=mod)x-=mod; while(x<0)x+=mod; return x;}
ll C(int n,int m)
{
  if(n<m)return 0;//!
  ll ret=1;
  for(int i=n-m+1;i<=n;i++)ret=(ret*i)%p;
  return ret;
}
void dfs(int x,int s,int t)
{
  if(x==n+1)
    {
      ans+=s*(C(n+b-t,n)-C(n+a-t-1,n)); ans=ans%p;
      return;
    }
  dfs(x+1,s,t);
  dfs(x+1,-s,t+m[x]+1);
}
int main()
{
  n=rd(); a=rd(); b=rd(); fac=1;
  for(int i=1;i<=n;i++)m[i]=rd(),fac*=i;
  p=(ll)fac*mod;
  dfs(1,1,0);
  /*
  for(int y=0;y<=b;y++)
    {
      ll tmp=upt(C(n+b-y,n)-C(n+a-y-1,n));
      ans=(ans+tmp*f[y])%p;
    }
  */
  if(ans<0)ans+=p;
  printf("%lld\n",ans/fac);
  return 0;
}