チーター本かTLE本に似た問題なかったっけ。
http://yukicoder.me/problems/no/489
問題
N日間の株価の遷移X[i]が与えられる。
ここで、株を一度にK個買い、それからD日以内にまとめて売る、ということを1回だけ行う。
利益の最大値と、その利益を達成するための株の売買日を答えよ。
解法
買う日iを総当たりし、それぞれ対応する売る日を求めよう。
i日目に購入した場合、X[i]~X[i+D]の間で最も株価が高い日に売ればよく、それには区間の最大値を求めるSegTreeを用いることができる。
その際、値だけでなくindexも求められるSegTreeを使おう。
なお、スライド最大値を使えばO(N)で解けるかも。
int N,D,K; int X[101010]; template<class V,int NV> class SegTree_Pair { public: vector<pair<V,int> > val; static V const def=0; pair<V,int> comp(pair<V,int> l,pair<V,int> r){ return max(l,r);} SegTree_Pair(){ val.resize(NV*2); int i; FOR(i,NV) val[i+NV]=make_pair(def,-i); for(i=NV-1;i>=1;i--) val[i]=comp(val[2*i],val[2*i+1]); }; pair<V,int> getval(int x,int y,int l=0,int r=NV,int k=1) { if(r<=x || y<=l) return make_pair(def,NV); if(x<=l && r<=y) return val[k]; return comp(getval(x,y,l,(l+r)/2,k*2),getval(x,y,(l+r)/2,r,k*2+1)); } void update(int entry, V v) { entry += NV; val[entry]=make_pair(v,-(entry-NV)); while(entry>1) entry>>=1, val[entry]=comp(val[entry*2],val[entry*2+1]); } }; SegTree_Pair<ll,1<<20> st; void solve() { int i,j,k,l,r,x,y; string s; cin>>N>>D>>K; FOR(i,N) { cin>>X[i]; st.update(i,X[i]); } ll dif=0; pair<int,int> P={0,0}; x=y=-1; FOR(i,N) { auto v=st.getval(i,i+D+1); if(dif < v.first-X[i]) { dif=v.first-X[i]; x=i; y=-v.second; } } cout<<dif*K<<endl; if(dif>0) cout<<x<<" "<<y<<endl; }
まとめ
株価を予測する能力欲しいですね。