Cod sursa(job #2002622)

Utilizator tziplea_stefanTiplea Stefan tziplea_stefan Data 20 iulie 2017 14:20:28
Problema 1-sir Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.54 kb
#include <fstream>
#define DEL 50000
#define MOD 194767

using namespace std;

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

int N, S, sum, i, j;
int a, b, dp[2][2*DEL];

int main()
{
    fin >> N >> S;
    a=dp[0][DEL]=1;
    for (i=1; i<N; i++)
    {
        sum+=i;
        for (j=DEL-sum; j<=DEL+sum; j++)
        {
            dp[a][j]=dp[b][j-i]+dp[b][j+i];
            if (dp[a][j]>=MOD)
              dp[a][j]-=MOD;
        }
        swap(a, b);
    }
    fout << dp[b][DEL+S] << '\n';
    fin.close();
    fout.close();
    return 0;
}