結果

問題 No.2025 Select $k$-th Submultiset
ユーザー 👑 potato167potato167
提出日時 2022-07-29 22:11:45
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 1,096 ms / 2,000 ms
コード長 2,528 bytes
コンパイル時間 2,237 ms
コンパイル使用メモリ 204,428 KB
実行使用メモリ 4,388 KB
最終ジャッジ日時 2023-09-26 21:15:15
合計ジャッジ時間 25,327 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 595 ms
4,384 KB
testcase_04 AC 547 ms
4,380 KB
testcase_05 AC 782 ms
4,380 KB
testcase_06 AC 810 ms
4,384 KB
testcase_07 AC 1,037 ms
4,384 KB
testcase_08 AC 1,004 ms
4,380 KB
testcase_09 AC 804 ms
4,384 KB
testcase_10 AC 845 ms
4,380 KB
testcase_11 AC 823 ms
4,384 KB
testcase_12 AC 878 ms
4,380 KB
testcase_13 AC 1,078 ms
4,380 KB
testcase_14 AC 1,096 ms
4,380 KB
testcase_15 AC 772 ms
4,380 KB
testcase_16 AC 910 ms
4,380 KB
testcase_17 AC 1,053 ms
4,384 KB
testcase_18 AC 804 ms
4,380 KB
testcase_19 AC 832 ms
4,380 KB
testcase_20 AC 758 ms
4,380 KB
testcase_21 AC 921 ms
4,380 KB
testcase_22 AC 919 ms
4,384 KB
testcase_23 AC 867 ms
4,384 KB
testcase_24 AC 1,022 ms
4,380 KB
testcase_25 AC 2 ms
4,380 KB
testcase_26 AC 1 ms
4,380 KB
testcase_27 AC 2 ms
4,384 KB
testcase_28 AC 2 ms
4,380 KB
testcase_29 AC 1 ms
4,380 KB
testcase_30 AC 2 ms
4,380 KB
testcase_31 AC 2 ms
4,384 KB
testcase_32 AC 2 ms
4,380 KB
testcase_33 AC 1 ms
4,384 KB
testcase_34 AC 2 ms
4,380 KB
testcase_35 AC 1 ms
4,388 KB
testcase_36 AC 3 ms
4,388 KB
testcase_37 AC 1 ms
4,384 KB
testcase_38 AC 1 ms
4,380 KB
testcase_39 AC 1 ms
4,380 KB
testcase_40 AC 2 ms
4,384 KB
testcase_41 AC 1 ms
4,384 KB
testcase_42 AC 2 ms
4,380 KB
testcase_43 AC 1 ms
4,380 KB
testcase_44 AC 2 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define _GLIBCXX_DEBUG
using namespace std;
using std::cout;
using std::cin;
using std::endl;
using ll=long long;
using ld=long double;
ll ILL=2167167167167167167;
const int INF=2100000000;
const ll mod=998244353;
#define rep(i,a) for (ll i=0;i<a;i++)
#define all(p) p.begin(),p.end()
template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>;
template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;}
template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;}
template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());}
template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});}
void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";}
template<class T> void vec_out(vector<T> &p){for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";}


void solve();
// oddloop
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int t=1;
	//cin>>t;
	rep(i,t) solve();
}

void solve(){
	int N,L;
	cin>>N>>L;
	auto comb=[](ll a,ll b)->ll{
		if(b<0) return 0;
		if(a<b) return 0;
		ll an=1;
		rep(i,b) an*=(a-i),an/=(i+1);
		return an;
	};
	vector<ll> p(N);
	rep(i,N) cin>>p[i];
	int Q;
	cin>>Q;
	if(N==1){
		rep(i,Q){
			ll k;
			cin>>k;
			if(k!=1) cout<<"-1\n";
			else{
				if(p[0]>=L) cout<<L<<"\n";
				else cout<<p[0]<<"\n";
			}
		}
		return ;
	}
	rep(i,Q){
		ll K;
		cin>>K;
		vector<int> ans(N);
		ll sum=0;
		rep(j,N-1){
			ll l=-1,r=p[j]+1;
			while(r-l>1){
				ll med=(l+r)/2;
				ll tmp=0;
				rep(k,1<<(N-j)){
					ll Z=L-sum;
					ll si=1;
					for(int x=j;x<N;x++){
						if(k&(1<<(x-j))){
							Z-=p[x]+1;
							si*=-1;
						}else{
							if(x==j) Z-=med;
						}
					}
					//cout<<k<<" "<<tmp<<"\n";
					tmp+=si*comb(Z+N-j-1,N-j-1);
				}
				//cout<<tmp<<" "<<med<<"#\n";
				if(tmp<K) r=med;
				else l=med;
			}
			//cout<<"# "<<k<<" , ";
			//cout<<l<<" "<<r<<" :\n";
			if(l==-1){
				ans[0]=-1;
				break;
			}
			ans[j]=l;
			sum+=l;
			rep(k,1<<(N-j)){
				ll Z=L-sum+l;
				ll si=1;
				for(int x=j;x<N;x++){
					if(k&(1<<(x-j))){
						Z-=p[x]+1;
						si*=-1;
					}else{
						if(x==j) Z-=r;
					}
				}
				K-=si*comb(Z+N-j-1,N-j-1);
			}
		}
		ans[N-1]=L-sum;
		if(ans[0]==-1) cout<<"-1\n";
		else vec_out(ans);
	}
}
0