Submission #318079

#	Time	Username	Problem	Language	Result	Execution time	Memory
318079		shivensinha4	Brunhilda’s Birthday (BOI13_brunhilda)	C++17	3.33 / 100	20 ms	3180 KiB

brunhilda

#include "bits/stdc++.h"
using namespace std; 
#define for_(i, s, e) for (int i = s; i < (int) e; i++)
#define for__(i, s, e) for (ll i = s; i < e; i++)
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> ii;
#define endl '\n'

const int MXV = 1e4;
int ans[MXV+2];
vi fac[MXV+2];

int main() {
	#ifdef shiven
	freopen("test.in", "r", stdin);
	#endif
	
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	
	int m, q; cin >> m >> q;
	int mxp = -1;
	for_(i, 0, m)  {
		int p; cin >> p;
		fac[p].push_back(p);
		mxp = max(mxp, p);
	}
	
	vector<ii> last; // {val, nxt}
	last.push_back({0, mxp});
	int pt = 0;
	for_(i, 1, MXV+1) {
		if (pt < last.size() and last[pt].second == i) pt++;
		
		if (pt < last.size()) {
			ans[i] = ans[last[pt].first]+1;
		}
		
		for_(f, 0, fac[i].size()) {
			int nxt = i+fac[i][f];
			if (ans[i] and last.back().second < nxt) last.push_back({i, nxt});
			if (nxt <= MXV) fac[nxt].push_back(f);
			// fac[i].pop_back();
		}
		
		// for (int f = fac[i].size()-1; f >= 0; f--) {
		// 	int nxt = i+fac[i][f];
		// 	if (ans[i] and last.back().second < nxt) last.push_back({i, nxt});
		// 	if (nxt <= MXV) fac[nxt].push_back(f);
		// 	// fac[i].pop_back();
		// }
	}
	
	for_(i, 0, q) {
		int k; cin >> k;
		// assert(k <= MXV);
		if (!ans[k]) cout << "oo";
		else cout << ans[k];
		cout << endl;
	}
	

	return 0;
}

Compilation message (stderr)

brunhilda.cpp: In function 'int main()':
brunhilda.cpp:34:10: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |   if (pt < last.size() and last[pt].second == i) pt++;
      |       ~~~^~~~~~~~~~~~~
brunhilda.cpp:36:10: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |   if (pt < last.size()) {
      |       ~~~^~~~~~~~~~~~~

• Subtask 1: 3.3333333333333335 / 20
• Subtask 2: 0 / 20
• Subtask 3: 0 / 60