Cod sursa(job #2854993)

Utilizator raileanu-alin-gabrielRaileanu Alin-Gabriel raileanu-alin-gabriel Data 21 februarie 2022 22:09:05
Problema 1-sir Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <fstream>
#define NMAX 258
#define MOD 194767
#define SMAX 32896

using namespace std;
ifstream fin("1-sir.in");
ofstream fout("1-sir.out");

int a[2][2*SMAX][2*NMAX], prec, crt, i, j, k, sum, n, s;

int main()
{
  prec=1;
  fin>>n>>s;
  a[prec][SMAX][NMAX]=1;
  for(i=2; i<=n; i++)
  {
    sum=(i*(i-1))/2;
    for(j=-sum; j<=sum; j++)
    {
      for(k=-i+1; k<i; k++)
        {
          a[crt][j+SMAX][k+NMAX]=(a[prec][j-k+SMAX][k-1+NMAX]+a[prec][j-k+SMAX][k+1+NMAX])%MOD;
          //fout<<i<<' '<<j<<' '<<k<<' '<<a[crt][j+SMAX][k+NMAX]<<'\n';
        }
    }
    prec=1-prec;
    crt=1-crt;
  }
  sum=0;
  for(k=-n+1; k<n; k++) sum=(sum+a[prec][s+SMAX][k+NMAX])%MOD;
  fout<<sum<<'\n';
  return 0;
}