#826517 (C++17) No.2161 Black Market

提出ソース

結果

問題	No.2161 Black Market
ユーザー	a16784542
提出日時	2022-12-17 17:09:11
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	170 ms / 7,000 ms
コード長	4,272 bytes
コンパイル時間	6,595 ms
コンパイル使用メモリ	308,684 KB
実行使用メモリ	17,108 KB
最終ジャッジ日時	2024-04-28 11:07:36
合計ジャッジ時間	9,201 ms
ジャッジサーバーID （参考情報）	judge2 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	2 ms 5,376 KB
testcase_13	AC	2 ms 5,376 KB
testcase_14	AC	2 ms 5,376 KB
testcase_15	AC	2 ms 5,376 KB
testcase_16	AC	2 ms 5,376 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	2 ms 5,376 KB
testcase_19	AC	3 ms 5,376 KB
testcase_20	AC	159 ms 16,976 KB
testcase_21	AC	156 ms 17,104 KB
testcase_22	AC	55 ms 9,964 KB
testcase_23	AC	151 ms 16,976 KB
testcase_24	AC	163 ms 17,104 KB
testcase_25	AC	96 ms 15,596 KB
testcase_26	AC	170 ms 17,108 KB
testcase_27	AC	45 ms 9,296 KB
testcase_28	AC	47 ms 9,296 KB
testcase_29	AC	15 ms 5,376 KB
testcase_30	AC	6 ms 5,376 KB
testcase_31	AC	106 ms 15,440 KB
testcase_32	AC	44 ms 9,296 KB
testcase_33	AC	12 ms 5,376 KB
testcase_34	AC	9 ms 5,376 KB
testcase_35	AC	12 ms 5,376 KB
testcase_36	AC	8 ms 5,376 KB
testcase_37	AC	4 ms 5,376 KB
testcase_38	AC	20 ms 5,712 KB
testcase_39	AC	4 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#if !__INCLUDE_LEVEL__

#include __FILE__

struct Solver {

int n,k,l,p;
V<TP> left;
V<V<P>> righ;
void inp(){
	cin>>n>>k>>l>>p;
	V<int> a1,a2,b1,b2;
	rep(i,n){
		int a,b;cin>>a>>b;
		a1.emplace_back(a);
		b1.emplace_back(b);
		swap(a1,a2);
		swap(b1,b2);
	}
	int x=sz(a1);
	rep(i,1<<x){
		int cost=0;
		int qua=0;
		int val=0;
		rep(j,x){
			if((i>>j)&1){
				cost+=a1[j];
				val+=b1[j];
				qua++;
			}
		}
		left.emplace_back(cost,qua,val);
	}

	x=sz(a2);
	righ.resize(x+1);
	rep(i,1<<x){
		int cost=0;
		int qua=0;
		int val=0;
		rep(j,x){
			if((i>>j)&1){
				cost+=a2[j];
				val+=b2[j];
				qua++;
			}
		}
		righ[qua].emplace_back(cost,val);
	}


	sort(all(left),greater<>());
	rep(i,sz(righ)){
		sort(all(righ[i]),greater<>());
	}

	return ;

}
void solve() {
	inp();

	ll ans=0;
	V<tree<P, null_type, less<P>, rb_tree_tag, tree_order_statistics_node_update>> s(sz(righ));
	int now=0;
	rep(i,sz(left)){
		auto [lc,lq,lv]=left[i];
		rep(j,sz(righ)){
			if(k<lq+j)break;
			while(sz(righ[j])){
				auto [rc,rv]=righ[j].back();
				if(l<lc+rc)break;
				s[j].insert(mpa(rv,now));
				now++;
				righ[j].pop_back();
			}
			// cout<<lc<<" "<<lq<<" "<<lv<<" "<<j<<" "<<s[j].order_of_key(p-lv)<<endl;
			ans+=sz(s[j])-s[j].order_of_key(mpa(p-lv,-1));
		}
	}

	cout<<ans<<endl;
	return ;
}};

signed main() {
	int ts = 1;
	// scanf("%lld",&ts);
	rep(ti,ts) {
		Solver solver;
		solver.solve();
	}
	return 0;
}

#else

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;

using namespace std;

#define int ll

using ll=long long;
using ld = long double;
using uint = unsigned;
using ull = unsigned long long;
using P=pair<int,int>;
using TP=tuple<int,int,int>;
template <class T> using V = vector<T>;
template <class T> using max_heap = priority_queue<T>;
template <class T> using min_heap = priority_queue<T, vector<T>, greater<>>;

#define FOR1(a) for (ll _ = 0; _ < ll(a); ++_)
#define FOR2(i, a) for (ll i = 0; i < ll(a); ++i)
#define FOR3(i, a, b) for (ll i = a; i < ll(b); ++i)
#define FOR4(i, a, b, c) for (ll i = a; i < ll(b); i += (c))
#define FOR1_R(a) for (ll i = (a)-1; i >= ll(0); --i)
#define FOR2_R(i, a) for (ll i = (a)-1; i >= ll(0); --i)
#define FOR3_R(i, a, b) for (ll i = (b)-1; i >= ll(a); --i)
#define FOR4_R(i, a, b, c) for (ll i = (b)-1; i >= ll(a); i -= (c))
#define overload4(a, b, c, d, e, ...) e
#define rep(...) overload4(__VA_ARGS__, FOR4, FOR3, FOR2, FOR1)(__VA_ARGS__)
#define rrep(...) overload4(__VA_ARGS__, FOR4_R, FOR3_R, FOR2_R, FOR1_R)(__VA_ARGS__)
#define fore(i,a) for(auto &i:a)

#define all(x) x.begin(),x.end() 
#define sz(x) ((int)(x).size())
#define bp(x) (__builtin_popcountll((long long)(x)))
#define elif else if
#define mpa make_pair
#define bit(n) (1LL<<(n))
#define LB(c, x) distance((c).begin(), lower_bound(all(c), (x)))
#define UB(c, x) distance((c).begin(), upper_bound(all(c), (x)))
#define UNIQUE(x) sort(all(x)), x.erase(unique(all(x)), x.end())

template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
template<typename T>string join(const T&v,const string& d=""){stringstream s;rep(i,sz(v))(i?s<<d:s)<<v[i];return s.str();}
template<typename T>ostream& operator<<(ostream&o,const vector<T>&v){if(sz(v))o<<join(v," ");return o;}
template<typename T1,typename T2>ostream& operator<<(ostream&o,const pair<T1,T2>&v){return o<<v.first<<","<<v.second;}
template<typename T1,typename T2>bool mins(T1& x,const T2&y){if(x>y){x=y;return true;}else return false;}
template<typename T1,typename T2>bool maxs(T1& x,const T2&y){if(x<y){x=y;return true;}else return false;}
template<typename Tx, typename Ty>Tx dup(Tx x, Ty y){return (x+y-1)/y;}
template<typename T>ll suma(const vector<T>&a){ll res(0);for(auto&&x:a)res+=x;return res;}
template<typename T>ll suma(const V<V<T>>&a){ll res(0);for(auto&&x:a)res+=suma(x);return res;}
template<typename T>void uni(T& a){sort(rng(a));a.erase(unique(rng(a)),a.end());}
template<typename T>void prepend(vector<T>&a,const T&x){a.insert(a.begin(),x);}

const int INF = 1001001001;
const ll INFL = 3e18;
const int MAX = 2e6+5;


#endif

yukicoder

結果

テストケース

ソースコード