Cod sursa(job #2948181)

Utilizator mihnea.cazan15mihnea cazan mihnea.cazan15 Data 27 noiembrie 2022 13:23:18
Problema 1-sir Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.64 kb
#include <fstream>

using namespace std;
ifstream fin("1-sir.in");
ofstream fout("1-sir.out");
int *dp[2],aux[2][65840];
const int mod=194767;
int main()
{
    int n,s;
    fin>>n>>s;
    dp[0]=aux[0]+n*(n-1)/2;
    dp[1]=aux[1]+n*(n-1)/2;
    dp[0][0]=1;
    for(int i=1;i<n;i++)
    {
        int curr=i%2;
        for(int j=0;j<65840;j++)
            aux[curr][j]=0;
        int l=i*(i+1)/2;
        for(int j=-l;j<=l-i;j++)
            dp[curr][j+i]=(dp[curr][j+i]+dp[1-curr][j])%mod;
        for(int j=-l+i;j<=l;j++)
            dp[curr][j-i]=(dp[curr][j-i]+dp[1-curr][j])%mod;
    }
    fout<<dp[1-n%2][s];
    return 0;
}