#266156 (C++14) No.704 ゴミ拾い Medium

提出ソース

結果

問題	No.704 ゴミ拾い Medium
ユーザー	しらっ亭
提出日時	2018-06-15 23:48:16
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	6,611 bytes
コンパイル時間	1,236 ms
コンパイル使用メモリ	120,640 KB
実行使用メモリ	29,020 KB
最終ジャッジ日時	2024-06-30 15:37:26
合計ジャッジ時間	6,331 ms
ジャッジサーバーID （参考情報）	judge2 / judge1

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	2 ms 6,944 KB
testcase_04	AC	2 ms 6,944 KB
testcase_05	AC	2 ms 6,944 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	2 ms 6,944 KB
testcase_10	WA	-
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	2 ms 6,940 KB
testcase_13	AC	2 ms 6,944 KB
testcase_14	WA	-
testcase_15	AC	2 ms 6,944 KB
testcase_16	AC	2 ms 6,940 KB
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	AC	129 ms 28,812 KB
testcase_25	AC	150 ms 28,884 KB
testcase_26	AC	138 ms 28,868 KB
testcase_27	AC	134 ms 28,816 KB
testcase_28	AC	144 ms 28,856 KB
testcase_29	AC	130 ms 28,888 KB
testcase_30	AC	135 ms 28,996 KB
testcase_31	AC	143 ms 28,848 KB
testcase_32	AC	133 ms 28,984 KB
testcase_33	AC	137 ms 29,020 KB
testcase_34	AC	137 ms 28,880 KB
testcase_35	AC	139 ms 28,816 KB
testcase_36	AC	139 ms 28,884 KB
testcase_37	AC	137 ms 28,824 KB
testcase_38	AC	134 ms 28,900 KB
testcase_39	AC	136 ms 28,848 KB
testcase_40	AC	135 ms 28,912 KB
testcase_41	AC	134 ms 28,856 KB
testcase_42	AC	137 ms 28,856 KB
testcase_43	AC	136 ms 28,884 KB
testcase_44	AC	2 ms 6,940 KB
testcase_45	AC	2 ms 6,944 KB
testcase_46	WA	-
testcase_47	WA	-

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

ソースコード

raw source code

#include <iostream>
#include <algorithm>
#include <bitset>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <utility>
#include <vector>
#include <array>
#include <unordered_map>
#include <complex>
#include <deque>
#include <cassert>
#include <cmath>
#include <functional>
#include <iomanip>
#include <chrono>
#include <random>
#include <numeric>
using namespace std;

#define forr(x,arr) for(auto&& x:arr)
#define _overload3(_1,_2,_3,name,...) name
#define _rep2(i,n) _rep3(i,0,n)
#define _rep3(i,a,b) for(int i=int(a);i<int(b);++i)
#define rep(...) _overload3(__VA_ARGS__,_rep3,_rep2,)(__VA_ARGS__)
#define _rrep2(i,n) _rrep3(i,0,n)
#define _rrep3(i,a,b) for(int i=int(b)-1;i>=int(a);i--)
#define rrep(...) _overload3(__VA_ARGS__,_rrep3,_rrep2,)(__VA_ARGS__)
#define all(x) (x).begin(),(x).end()
#define bit(n) (1LL<<(n))
#define sz(x) ((int)(x).size())
#define TEN(n) ((ll)(1e##n))
#define fst first
#define snd second

string DBG_DLM(int &i){return(i++==0?"":", ");}
#define DBG_B(exp){int i=0;os<<"{";{exp;}os<<"}";return os;}
template<class T>ostream&operator<<(ostream&os,vector<T>v);
template<class T>ostream&operator<<(ostream&os,set<T>v);
template<class T>ostream&operator<<(ostream&os,queue<T>q);
template<class T>ostream&operator<<(ostream&os,priority_queue<T>q);
template<class T,class K>ostream&operator<<(ostream&os,pair<T,K>p);
template<class T,class K>ostream&operator<<(ostream&os,map<T,K>mp);
template<class T,class K>ostream&operator<<(ostream&os,unordered_map<T,K>mp);
template<int I,class TPL>void DBG(ostream&os,TPL t){}
template<int I,class TPL,class H,class...Ts>void DBG(ostream&os,TPL t){os<<(I==0?"":", ")<<get<I>(t);DBG<I+1,TPL,Ts...>(os,t);}
template<class T,class K>void DBG(ostream&os,pair<T,K>p,string delim){os<<"("<<p.first<<delim<<p.second<<")";}
template<class...Ts>ostream&operator<<(ostream&os,tuple<Ts...>t){os<<"(";DBG<0,tuple<Ts...>,Ts...>(os,t);os<<")";return os;}
template<class T,class K>ostream&operator<<(ostream&os,pair<T,K>p){DBG(os,p,", ");return os;}
template<class T>ostream&operator<<(ostream&os,vector<T>v){DBG_B(forr(t,v){os<<DBG_DLM(i)<<t;});}
template<class T>ostream&operator<<(ostream&os,set<T>s){DBG_B(forr(t,s){os<<DBG_DLM(i)<<t;});}
template<class T>ostream&operator<<(ostream&os,queue<T>q){DBG_B(for(;q.size();q.pop()){os<<DBG_DLM(i)<<q.front();});}
template<class T>ostream&operator<<(ostream&os,priority_queue<T>q){DBG_B(for(;q.size();q.pop()){os<<DBG_DLM(i)<<q.top();});}
template<class T,class K>ostream&operator<<(ostream&os,map<T,K>m){DBG_B(forr(p,m){os<<DBG_DLM(i);DBG(os,p,"->");});}
template<class T,class K>ostream&operator<<(ostream&os,unordered_map<T,K>m){DBG_B(forr(p,m){os<<DBG_DLM(i);DBG(os,p,"->");});}
#define DBG_OVERLOAD(_1,_2,_3,_4,_5,_6,macro_name,...)macro_name
#define DBG_LINE(){char s[99];sprintf(s,"line:%3d | ",__LINE__);cerr<<s;}
#define DBG_OUTPUT(v){cerr<<(#v)<<"="<<(v);}
#define DBG1(v,...){DBG_OUTPUT(v);}
#define DBG2(v,...){DBG_OUTPUT(v);cerr<<", ";DBG1(__VA_ARGS__);}
#define DBG3(v,...){DBG_OUTPUT(v);cerr<<", ";DBG2(__VA_ARGS__);}
#define DBG4(v,...){DBG_OUTPUT(v);cerr<<", ";DBG3(__VA_ARGS__);}
#define DBG5(v,...){DBG_OUTPUT(v);cerr<<", ";DBG4(__VA_ARGS__);}
#define DBG6(v,...){DBG_OUTPUT(v);cerr<<", ";DBG5(__VA_ARGS__);}
#define DEBUG0(){DBG_LINE();cerr<<endl;}
#ifdef LOCAL
#define out(...){DBG_LINE();DBG_OVERLOAD(__VA_ARGS__,DBG6,DBG5,DBG4,DBG3,DBG2,DBG1)(__VA_ARGS__);cerr<<endl;}
#else
#define out(...)
#endif

using ll=long long;
using pii=pair<int,int>;using pll=pair<ll,ll>;using pil=pair<int,ll>;using pli=pair<ll,int>;
using vs=vector<string>;using vvs=vector<vs>;using vvvs=vector<vvs>;
using vb=vector<bool>;using vvb=vector<vb>;using vvvb=vector<vvb>;
using vi=vector<int>;using vvi=vector<vi>;using vvvi=vector<vvi>;
using vl=vector<ll>;using vvl=vector<vl>;using vvvl=vector<vvl>;
using vd=vector<double>;using vvd=vector<vd>;using vvvd=vector<vvd>;
using vpii=vector<pii>;using vvpii=vector<vpii>;using vvvpii=vector<vvpii>;
template<class A,class B>bool amax(A&a,const B&b){return b>a?a=b,1:0;}
template<class A,class B>bool amin(A&a,const B&b){return b<a?a=b,1:0;}
template<class T>void print(T t){cout<<t<<"\n";}
template<class H,class... T>void print(H h,T... t){cout<<h<<" ";print(t...);}
ll ri(){ll l;cin>>l;return l;} string rs(){string s;cin>>s;return s;}

template<class V, class Merge> struct SegmentTree {
	const int n;
	const V unit_value;
	vector<V> val;

	SegmentTree(int _n) : n(1 << (32-__builtin_clz(_n-1))), unit_value(Merge::unit()), val(n * 2, unit_value) {}

	V get(int i) const { return val[i + n]; }
	void set(int i, const V &v) { val[i + n] = v; }

	void build() {
		for (int i = n - 1; i > 0; i--) val[i] = Merge::merge(val[i << 1], val[i << 1 | 1]);
	}

	void update(int i, const V &v) {
		for (val[i += n] = v; i >>= 1;) {
			val[i] = Merge::merge(val[i << 1], val[i << 1 | 1]);
		}
	}

	V query(int l, int r) const {
		l = max(0, min(n, l)) + n;
		r = max(0, min(n, r)) + n;
		V ret1 = unit_value, ret2 = unit_value;
		for (; l < r; l >>= 1, r >>= 1) {
			if (l & 1) ret1 = Merge::merge(ret1, val[l++]);
			if (r & 1) ret2 = Merge::merge(val[--r], ret2);
		}
		return Merge::merge(ret1, ret2);
	}
};
template<class V, class Merge> ostream& operator<<(ostream& os, const SegmentTree<V, Merge>& seg) {
	vector<V> vec;
	for (int i = 0; i < seg.n; i++) {
		vec.emplace_back(seg.get(i));
	}
	os << vec;
	return os;
}

template <class V> struct MergeRangeMinQ {
	static V merge(const V &l, const V &r) { return l < r ? l : r; }
	static V unit() { return numeric_limits<V>::max(); }
};
template<class V> using SegTreeMin = SegmentTree<V, MergeRangeMinQ<V>>;

ll solve() {
	int n = ri();
	vector<ll> A(n);
	rep(i, n) A[i] = ri();
	vector<ll> X(n);
	rep(i, n) X[i] = ri();
	vector<ll> Y(n);
	rep(i, n) Y[i] = ri();

	const ll inf = 1e18;

	vl dp(n+1, inf);
	dp[0] = 0;

	if (0) {
		auto cost = [&](int i, int j) -> ll {
			ll dx = abs(X[j] - A[i]);
			ll dy = Y[j];
			return dx + dy;
		};

		rep(i, n) {
			rep(j, i+1) {
				amin(dp[i+1], dp[j] + cost(i, j));
			}
		}
		out(dp);
	}

	fill(all(dp), inf);
	dp[0] = 0;

	SegTreeMin<ll> seg1(n), seg2(n);

	rep(i, n) {
		seg1.update(i, dp[i] + X[i] + Y[i]);
		seg2.update(i, dp[i] - X[i] + Y[i]);

		int idx = lower_bound(X.begin(), X.end(), A[i]) - X.begin();
		out(i, A[i], idx);
		ll cand1 = +A[i] + seg2.query(0, idx);
		ll cand2 = -A[i] + seg1.query(idx, i+1);
		out(cand1, cand2);

		dp[i+1] = min(cand1, cand2);
	}
	out(dp);

	return dp[n];

}

void Main() {
	cout << solve() << endl;
}
signed main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	Main();
	return 0;
}

yukicoder

結果

テストケース

ソースコード