結果
問題 | No.2741 Balanced Choice |
ユーザー |
![]() |
提出日時 | 2024-04-20 10:44:07 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 239 ms / 2,000 ms |
コード長 | 10,161 bytes |
コンパイル時間 | 3,215 ms |
コンパイル使用メモリ | 228,920 KB |
最終ジャッジ日時 | 2025-02-21 06:10:59 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 10 |
ソースコード
// author: hotman78// date: 2024/04/20-10:44:00// --- begin raw code -----------------// #include"cpplib/util/template.hpp"// // #include"cpplib/graph_tree/dijkstra.hpp"//// void solve(){// lint n,w,d;// cin>>n>>w>>d;// // 部分和問題を分けて解いて// vector<pair<lint,lint>>v1;// vector<pair<lint,lint>>v2;// rep(i,n){// lint t,v,w;// cin>>t>>w>>v;// if(t==0)v1.emplace_back(w,v);// if(t==1)v2.emplace_back(w,v);// }// auto dp1=make_vector<lint>(-INF,v1.size()+1,w+1);// auto dp2=make_vector<lint>(-INF,v2.size()+1,w+1);// dp1[0][0]=dp2[0][0]=0;// rep(i,v1.size()){// rep(j,w+1){// if(j+v1[i].first<=w){// chmax(dp1[i+1][j+v1[i].first],dp1[i][j]+v1[i].second);// }// chmax(dp1[i+1][j],dp1[i][j]);// }// }// rep(i,v2.size()){// rep(j,w+1){// if(j+v2[i].first<=w){// chmax(dp2[i+1][j+v2[i].first],dp2[i][j]+v2[i].second);// }// chmax(dp2[i+1][j],dp2[i][j]);//// }// }// // debug(dp1[v1.size()]);// lint ans=0;// rep(i,w+1)rep(j,w+1){// if(i+j>w)continue;// if(abs(i-j)>d)continue;// chmax(ans,dp1[v1.size()][i]+dp2[v2.size()][j]);// }// cout<<ans<<endl;// }//// int main(){// solve();// // lint t;cin>>t;while(t--)solve();// }// --- end raw code -----------------#line 2 "cpplib/util/template.hpp"#ifdef LOCAL#define _GLIBCXX_DEBUG#endif#pragma GCC optimize("Ofast")#pragma GCC optimize("unroll-loops")// #pragma GCC target("avx2")#include <bits/stdc++.h>using namespace std;#line 1 "cpplib/util/ioutil.hpp"// template <class Head,class... Args>// std::ostream& output(std::ostream& out,const Head& head,const Args&... args){// out>>head;// return output(head,args...);// }// template <class Head>// std::ostream& output(std::ostream& out,const Head& head){// out>>head;// return out;// }template <typename T, typename E>std::ostream &operator<<(std::ostream &out, std::pair<T, E> v) {out << "(" << v.first << "," << v.second << ")";return out;}// template <class... Args>// ostream& operator<<(ostream& out,std::tuple<Args...>v){// std::apply(output,v);// return out;// }#line 11 "cpplib/util/template.hpp"struct __INIT__ {__INIT__() {cin.tie(0);ios::sync_with_stdio(false);cout << fixed << setprecision(15);}} __INIT__;typedef long long lint;constexpr long long INF = 1LL << 60;constexpr int IINF = 1 << 30;constexpr double EPS = 1e-10;#ifndef REACTIVE#define endl '\n';#endiftypedef vector<lint> vec;typedef vector<vector<lint>> mat;typedef vector<vector<vector<lint>>> mat3;typedef vector<string> svec;typedef vector<vector<string>> smat;template <typename T> using V = vector<T>;template <typename T> using VV = V<V<T>>;#define output(t) \{ \bool f = 0; \for (auto val : (t)) { \cout << (f ? " " : "") << val; \f = 1; \} \cout << endl; \}#define output2(t) \{ \for (auto i : t) \output(i); \}#define debug(t) \{ \bool f = 0; \for (auto i : t) { \cerr << (f ? " " : "") << i; \f = 1; \} \cerr << endl; \}#define debug2(t) \{ \for (auto i : t) \debug(i); \}#define loop(n) for (long long _ = 0; _ < (long long)(n); ++_)#define _overload4(_1, _2, _3, _4, name, ...) name#define __rep(i, a) repi(i, 0, a, 1)#define _rep(i, a, b) repi(i, a, b, 1)#define repi(i, a, b, c) \for (long long i = (long long)(a); i < (long long)(b); i += c)#define rep(...) _overload4(__VA_ARGS__, repi, _rep, __rep)(__VA_ARGS__)#define _overload3_rev(_1, _2, _3, name, ...) name#define _rep_rev(i, a) repi_rev(i, 0, a)#define repi_rev(i, a, b) \for (long long i = (long long)(b)-1; i >= (long long)(a); --i)#define rrep(...) _overload3_rev(__VA_ARGS__, repi_rev, _rep_rev)(__VA_ARGS__)#define all(n) begin(n), end(n)template <typename T, typename E> bool chmin(T &s, const E &t) {bool res = s > t;s = min<T>(s, t);return res;}template <typename T, typename E> bool chmax(T &s, const E &t) {bool res = s < t;s = max<T>(s, t);return res;}const vector<lint> dx = {1, 0, -1, 0, 1, 1, -1, -1};const vector<lint> dy = {0, 1, 0, -1, 1, -1, 1, -1};#define SUM(v) accumulate(all(v), 0LL)#if __cplusplus >= 201703Ltemplate <typename T, typename... Args>auto make_vector(T x, int arg, Args... args) {if constexpr (sizeof...(args) == 0)return vector<T>(arg, x);elsereturn vector(arg, make_vector<T>(x, args...));}#endif#define bit(n, a) ((n >> a) & 1)#define extrep(v, ...) for (auto v : make_mat_impl({__VA_ARGS__}))vector<vector<long long>> make_mat_impl(vector<long long> v) {if (v.empty())return vector<vector<long long>>(1, vector<long long>());long long n = v.back();v.pop_back();vector<vector<long long>> ret;vector<vector<long long>> tmp = make_mat_impl(v);for (auto e : tmp)for (long long i = 0; i < n; ++i) {ret.push_back(e);ret.back().push_back(i);}return ret;}using graph = vector<vector<int>>;template <typename T> using graph_w = vector<vector<pair<int, T>>>;#if __cplusplus >= 201703Lconstexpr inline long long powll(long long a, long long b) {long long res = 1;while (b--)res *= a;return res;}#endiftemplate <typename T, typename E>pair<T, E> &operator+=(pair<T, E> &s, const pair<T, E> &t) {s.first += t.first;s.second += t.second;return s;}template <typename T, typename E>pair<T, E> &operator-=(pair<T, E> &s, const pair<T, E> &t) {s.first -= t.first;s.second -= t.second;return s;}template <typename T, typename E>pair<T, E> operator+(const pair<T, E> &s, const pair<T, E> &t) {auto res = s;return res += t;}template <typename T, typename E>pair<T, E> operator-(const pair<T, E> &s, const pair<T, E> &t) {auto res = s;return res -= t;}#define BEGIN_STACK_EXTEND(size) \void *stack_extend_memory_ = malloc(size); \void *stack_extend_origin_memory_; \char *stack_extend_dummy_memory_ = (char *)alloca( \(1 + (int)(((long long)stack_extend_memory_) & 127)) * 16); \*stack_extend_dummy_memory_ = 0; \asm volatile("mov %%rsp, %%rbx\nmov %%rax, %%rsp" \: "=b"(stack_extend_origin_memory_) \: "a"((char *)stack_extend_memory_ + (size)-1024));#define END_STACK_EXTEND \asm volatile("mov %%rax, %%rsp" ::"a"(stack_extend_origin_memory_)); \free(stack_extend_memory_);int floor_pow(int n) { return n ? 31 - __builtin_clz(n) : 0; }#line 2 "main.cpp"// #include"cpplib/graph_tree/dijkstra.hpp"void solve() {lint n, w, d;cin >> n >> w >> d;// 部分和問題を分けて解いてvector<pair<lint, lint>> v1;vector<pair<lint, lint>> v2;rep(i, n) {lint t, v, w;cin >> t >> w >> v;if (t == 0)v1.emplace_back(w, v);if (t == 1)v2.emplace_back(w, v);}auto dp1 = make_vector<lint>(-INF, v1.size() + 1, w + 1);auto dp2 = make_vector<lint>(-INF, v2.size() + 1, w + 1);dp1[0][0] = dp2[0][0] = 0;rep(i, v1.size()) {rep(j, w + 1) {if (j + v1[i].first <= w) {chmax(dp1[i + 1][j + v1[i].first], dp1[i][j] + v1[i].second);}chmax(dp1[i + 1][j], dp1[i][j]);}}rep(i, v2.size()) {rep(j, w + 1) {if (j + v2[i].first <= w) {chmax(dp2[i + 1][j + v2[i].first], dp2[i][j] + v2[i].second);}chmax(dp2[i + 1][j], dp2[i][j]);}}// debug(dp1[v1.size()]);lint ans = 0;rep(i, w + 1) rep(j, w + 1) {if (i + j > w)continue;if (abs(i - j) > d)continue;chmax(ans, dp1[v1.size()][i] + dp2[v2.size()][j]);}cout << ans << endl;}int main() {solve();// lint t;cin>>t;while(t--)solve();}