Cod sursa(job #2716822)

Utilizator Alex_tz307Lorintz Alexandru Alex_tz307 Data 5 martie 2021 18:40:36
Problema 1-sir Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.62 kb
#include <bits/stdc++.h>
#define ABS(x) ((x) >= 0 ? (x) : -(x))

using namespace std;

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

const int mod = 194767;
int N, K, dp[33000];

void add_self(int &a, int b) {
    a += b;
    if(a >= mod)
        a -= mod;
}

int main() {
    fin >> N >> K;
    K = ABS(K);
    const int max_sum = N * (N - 1) / 2;
    if(K > max_sum) {
        fout << "0\n";
        return 0;
    }
    dp[0] = 1;
    for(int last = 1; last < N; ++last)
        for(int sum = K - last; sum >= 0; --sum)
            add_self(dp[sum + last], dp[sum]);
    fout << dp[K] << '\n';
}