AT3576 E Popping Balls——计数思路

题目:https://code-festival-2017-qualb.contest.atcoder.jp/tasks/code_festival_2017_qualb_e

题解:https://www.cnblogs.com/Miracevin/p/10963431.html

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=2005,mod=1e9+7;
int upt(int x){while(x>=mod)x-=mod;while(x<0)x+=mod;return x;}

int A,B,c[N][N],s[N][N],ans;
void init()
{
  int lm=max(A,B);
  for(int i=0;i<=lm;i++)c[i][0]=1;
  for(int i=1;i<=lm;i++)
    for(int j=1;j<=i;j++)
      c[i][j]=upt(c[i-1][j-1]+c[i-1][j]);
  for(int i=0;i<=lm;i++)
    for(int j=0;j<=lm;j++)
      s[i][j]=c[i][j];
  for(int i=0;i<=lm;i++)
    for(int j=0;j<=lm;j++)
      s[i][j]=upt(s[i][j-1]+s[i][j]);
  for(int i=0;i<=lm;i++)
    for(int j=0;j<=lm;j++)
      s[i][j]=upt(s[i][j-1]+s[i][j]);
}
int main()
{
  scanf("%d%d",&A,&B); init();
  for(int i=0;i<=A;i++)
    for(int t=0;t<=A-i;t++)
      {
    int tp=0;
    if(t)tp=s[t-1][A-i-t];
    else tp=1;
    ans=(ans+(ll)tp*c[B-1][t])%mod;
      }
  printf("%d\n",ans);
  return 0;
}

 

posted on 2019-06-13 17:00  Narh  阅读(190)  评论(0编辑  收藏  举报

导航