Cod sursa(job #2680365)

Utilizator PatrickCplusplusPatrick Kristian Ondreovici PatrickCplusplus Data 3 decembrie 2020 12:49:22
Problema 1-sir Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.72 kb
#include <bits/stdc++.h>

using namespace std;

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

const int mod = 194767;
int n, s;
int main(){
    fin >> n >> s;
    int maxim = n * (n + 1) / 2;
    if (s > maxim || s < -maxim){
        fout << 0;
        return 0;
    }
    vector <int> dp[2];
    dp[0].resize(4 * maxim + n + 5);
    dp[1].resize(4 * maxim + n + 5);
    dp[n % 2][s + maxim] = 1;
    for (int i = n - 1; i >= 1; --i){
        for (int j = -maxim; j <= maxim; ++j){
            dp[i % 2][j + maxim] = (dp[(i + 1) % 2][j + (n - i) + maxim] + dp[(i + 1) % 2][j - (n - i) + maxim]) % mod;
        }
    }
    fout << dp[1][maxim];
    fin.close();
    fout.close();
    return 0;
}