Submission #416690

#	Time	Username	Problem	Language	Result	Execution time	Memory
416690		inksamurai	Weird Numeral System (CCO21_day1problem2)	C++17	0 / 25	648 ms	1048580 KiB

Main

#include <cmath>
#include <deque>
#include <algorithm>
#include <iterator>
#include <list>
#include <tuple>
#include <map>
#include <unordered_map>
#include <queue>
#include <set>
#include <unordered_set>
#include <stack>
#include <string>
#include <vector>
#include <fstream>
#include <iostream>
#include <functional>
#include <numeric>
#include <iomanip> 
#include <stdio.h>
//eolibraries
#define lnf 3999999999999999999
#define inf 999999999
#define PI 3.14159265359
#define endl "\n"
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define ll long long
#define all(c) (c).begin(),(c).end()
#define sz(c) (int)(c).size()
#define mkp(a,b) make_pair(a,b)
#define make_unique(a) sort(all(a)),a.erase(unique(all(a)),a.end())
#define rsz(a,n) a.resize(n)
#define pii pair <int,int>
#define rep(i,n) for(int i = 0 ; i < n ; i++) 
#define drep(i,n) for(int i = n-1 ; i >= 0 ; i--)
#define crep(i,x,n) for(int i = x ; i < n ; i++)
#define vi vector <int> 
#define vec(s) vector<s>
#define rsz(a,n) a.resize(n)
#define rszv(a,n,v) a.resize(n,v)
#define fcin ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0);
//eodefine
 
const int max_n = 103002;
 
using namespace std;

int k,n,q,m;
vi a;


map <ll,bool> memo;

bool slv(ll x){

	if(memo.find(x)!=memo.end()) return memo[x];
	
	rep(i,n) {
		if(x-a[i] == 0) {
			// cout << x << " " << a[i] << "\n";
			return memo[x] = 1;
		}
		if((x-a[i])%k!=0) continue;
		ll nex=(x-a[i])/k;
		if(slv(nex) == 1) {
			// cout << x << " " << a[i] << " " << nex <<  "\n";
			return memo[x] = 1;
		}	
	}
	return memo[x]=0;
}

int main(){
fcin;
	cin>>k>>q>>n>>m;
	rep(i,n){int x;cin>>x;a.pb(x);}

	rep(_,q) {
		ll val;
		cin>>val;
		if(slv(val)==0) cout << "IMPOSSIBLE\n";
		else {
			vi tmp;
			while(val!=inf) {
				rep(i,sz(a)) {
					if(val==a[i]) {
						tmp.pb(a[i]);
						val=inf;break;
					}
					if((val-a[i])%k!=0)continue;
					if(memo[((val-a[i])/k)]==1) {val-=a[i];val/=k;tmp.pb(a[i]);}
				
				}
			}
			reverse(all(tmp));
			for(auto x:tmp)cout<<x<<" ";
			cout<<"\n";
		}
	}
 
/*
*/
 
    return 0;   
}

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