Cod sursa(job #2750658)

Utilizator bent_larsenSturzu Antonio-Gabriel bent_larsen Data 12 mai 2021 18:08:02
Problema 1-sir Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.69 kb
#include <iostream>
#include <cstring>
#include <fstream>
#include <map>
#include <cmath>
using namespace std;

map<pair<int,int>, int> mp;
const int mod = 194767;

int solve(int S, int N)
{
	if(abs(S) > (N * (N - 1)) / 2)
		return 0;
	if(N == 1)
	{
		if(!S)
			return 1;
		return 0;
	}
	pair<int,int> p = make_pair(S, N);
	if(mp.find(p) != mp.end())
		return mp[p];
	long long ans = 0;
	
	ans += solve(S - N + 1, N - 1);
	ans %= mod;
	ans += solve(S + N - 1, N - 1);
	ans %= mod;
	return mp[p] = ans;
}

int main() {
	ifstream in("1-sir.in");
	ofstream out("1-sir.out");
	int N, S;
	
	in >> N >> S;
	out << solve(S, N) << "\n";
	in.close();
	out.close();
	return 0;
}