Submission #1103706

#	Time	Username	Problem	Language	Result	Execution time	Memory
1103706		vjudge1	Weird Numeral System (CCO21_day1problem2)	C++17	25 / 25	1919 ms	6372 KiB

Main

#include <bits/stdc++.h>
using namespace std;
#define int long long

const long long inf = 2e18;

map<long long, long long> par;
int a[5008];

int d, K;
bool attempt(long long R) {
	if (par.count(R)) return (par[R] != inf);
	par[R] = inf;
	for (int i = 1; i <= d; ++i) if (abs(R - a[i]) % K == 0) {
		if (attempt((R - a[i]) / K)) {par[R] = a[i]; return 1;}
	}
	return 0;
}

signed main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	int q, m; cin >> K >> q >> d >> m;
	for (int i = 1; i <= d; ++i) cin >> a[i], par[a[i]] = a[i];
	long long N, CN;
	while (q--) {
		cin >> N; CN = N; 
		if (!attempt(N)) cout << "IMPOSSIBLE\n";
		else {
			vector<int> ans; long long x;
			while (1) {
				x = par[N]; ans.push_back(x);
				N = (N - x) / K;
				if (N == 0) break;
			}
			reverse(ans.begin(), ans.end());
			for (int i = 0; i + 1 < ans.size(); ++i) cout << ans[i] << ' ';
			cout << ans.back() << '\n';
		}
	}
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:37:26: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |    for (int i = 0; i + 1 < ans.size(); ++i) cout << ans[i] << ' ';
      |                    ~~~~~~^~~~~~~~~~~~
Main.cpp:25:15: warning: variable 'CN' set but not used [-Wunused-but-set-variable]
   25 |  long long N, CN;
      |               ^~

• Subtask 1: 8 / 8
• Subtask 2: 17 / 17