Cod sursa(job #2021061)

Utilizator Theodor1000Cristea Theodor Stefan Theodor1000 Data 12 septembrie 2017 17:12:01
Problema 1-sir Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <cstdio>
#include <algorithm>

using namespace std;

long long dp[2][140000];

int main ()
{
    freopen ("1-sir.in", "r", stdin);
    freopen ("1-sir.out", "w", stdout);

    int n, sum;
    scanf ("%d %d", &n, &sum);
    sum += 70000;

    int mi = -n * (n + 1) / 2 + 70000;
    int ma = n * (n + 1) / 2 + 70000;

    if (sum > ma || sum < mi)
    {
        printf ("0\n");
        return 0;
    }

    dp[0][70000] = 1LL;

    for (int i = 1; i < n; ++i)
    {
        for (int j = mi; j <= ma; ++j)
        {
            if (j + i <= ma) dp[1][j + i] += dp[0][j];
            if (j - i >= mi) dp[1][j - i] += dp[0][j];
        }

        for (int j = mi; j <= ma; ++j)
        {
            dp[0][j] = dp[1][j],
            dp[1][j] = 0LL;
        }
    }

    printf ("%lld\n", dp[0][sum]);

    return 0;
}