Cod sursa(job #3332109)

Utilizator andrei22116Popescu Stefan Andrei andrei22116 Data 4 ianuarie 2026 13:59:40
Problema 1-sir Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.68 kb
//#pragma GCC optimize("O3, Ofast, unroll-loops")
#include <bits/stdc++.h>
using namespace std;

const int MOD=194767;
vector<map<int,map<int,int>>> mat;

int main()
{
    ifstream cin("1-sir.in");
    ofstream cout("1-sir.out");
    int n,s,sum=0;
    cin >> n >> s;
    mat.resize(n);
    mat[0][0][0]=1;
    for(int i=1;i<n;i++)
    {
        for(int j=-i;j<=i;j++)
        {
            for(int k=-i*(i+1)/2;k<=i*(i+1)/2;k++)
            {
                mat[i][j][k]=(mat[i-1][j-1][k-j]+mat[i-1][j+1][k-j])%MOD;
            }
        }
    }
    for(int j=-n;j<=n;j++)
    {
        sum+=mat[n-1][j][s];
        sum%=MOD;
    }
    cout << sum;
    return 0;
}