# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
553057 | 2022-04-24T14:07:47 Z | mars4 | Split the sequence (APIO14_sequence) | C++17 | 865 ms | 81852 KB |
#include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace std; using namespace __gnu_pbds; #define ff first #define ss second #define ll int64_t #define ld long double #define nl cout<<"\n" #define i128 __int128_t #define all(v) v.begin(),v.end() #define mset(a,v) memset((a),(v),sizeof(a)) #define forn(i,a,b) for(int64_t i=int64_t(a);i<int64_t(b);++i) #define forb(i,a,b) for(int64_t i=int64_t(a);i>=int64_t(b);--i) #define fastio() ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); #define mod 1'000'000'007 #define mod2 998'244'353 #define inf 1'000'000'000'000'007 #define pi 3.14159265358979323846 template<class key,class cmp=std::less<key>> using ordered_set=tree<key,null_type,cmp,rb_tree_tag,tree_order_statistics_node_update>; template<class L,class R> ostream& operator<<(ostream& out,pair<L,R> &p) {return out<<"("<<p.ff<<", "<<p.ss<<")";} template<class T> ostream& operator<<(ostream& out,vector<T> &v) {out<<"[";for(auto it=v.begin();it!=v.end();++it){if(it!=v.begin())out<<", ";out<<*it;}return out<<"]";} template<class T> ostream& operator<<(ostream& out,deque<T> &v) {out<<"[";for(auto it=v.begin();it!=v.end();++it){if(it!=v.begin())out<<", ";out<<*it;}return out<<"]";} template<class T> ostream& operator<<(ostream& out,set<T> &s) {out<<"{";for(auto it=s.begin();it!=s.end();++it){if(it!=s.begin())out<<", ";out<<*it;}return out<<"}";} template<class T> ostream& operator<<(ostream& out,ordered_set<T> &s) {out<<"{";for(auto it=s.begin();it!=s.end();++it){if(it!=s.begin())out<<", ";out<<*it;}return out<<"}";} template<class L,class R> ostream& operator<<(ostream& out,map<L,R> &m) {out<<"{";for(auto it=m.begin();it!=m.end();++it){if(it!=m.begin())out<<", ";out<<*it;}return out<<"}";} void dbg_out() {cerr<<"]\n";} template<typename Head,typename... Tail> void dbg_out(Head H,Tail... T) {cerr<<H;if(sizeof...(Tail))cerr<<", ";dbg_out(T...);} #ifdef LOCAL #define dbg(...) cerr<<"["<<#__VA_ARGS__<<"] = [",dbg_out(__VA_ARGS__) #else #define dbg(...) #endif //---------------------------------mars4--------------------------------- const ll N=100'000; const ll K=200; ll a[N]; ll pre[N+1]; int v[K][N]; vector<ll> dp1; vector<ll> dp2; ll sum(ll l,ll r) { return pre[r+1]-pre[l]; } class CHT { struct Line { ll m; ll c; ll ind; Line(ll m,ll c,ll ind):m(m),c(c),ind(ind) {} ll eval(ll x) { return m*x+c; } }; public: deque<Line> cv_hull; void update(ll m,ll c,ll ind) { Line l=Line(m,c,ind); ll n=(ll)cv_hull.size(); while(n>=2 and (l.c-cv_hull[0].c)*(cv_hull[1].m-cv_hull[0].m)>=(cv_hull[0].c-cv_hull[1].c)*(cv_hull[0].m-l.m)) { cv_hull.pop_front(); n--; } cv_hull.push_front(l); } pair<ll,ll> query(ll x) { ll n=(ll)cv_hull.size(); while(n>=2 and cv_hull[n-1].eval(x)<=cv_hull[n-2].eval(x)) { cv_hull.pop_back(); n--; } return {cv_hull[n-1].eval(x),cv_hull[n-1].ind}; } }; int main() { fastio(); ll z,n,m,t,k,i,j,l,d,h,r; cin>>n>>k; dp1=vector<ll>(n); dp2=vector<ll>(n); forn(i,0,n) { cin>>a[i]; pre[i+1]=pre[i]+a[i]; } // dp[i][k]=dp[j][k]+(sum(j+1,n-1)-sum(r+1,n-1))*sum(r+1,n-1) // dp[i][k]=dp[j][k-1]+sum(j+1,n-1)*sum(r+1,n-1)-sum(r+1,n-1)*sum(r+1,n-1); // y=mx+c // m=sum(j+1,n-1), x=sum(r+1,n-1) // m is decreasing, x is decreasing forn(j,0,k) { CHT cht; forn(i,j,n-1) { cht.update(sum(i,n-1),i?dp1[i-1]:0,i); ll rsum=sum(i+1,n-1); auto [val,ind]=cht.query(rsum); dp2[i]=val-rsum*rsum; v[j][i]=ind; } dp1.swap(dp2); } pair<ll,ll> mval={-1,-1}; forn(i,0,n-1) { mval=max(mval,{dp1[i],i}); } vector<ll> ans; ll ind=mval.ss; forb(i,k-1,0) { ans.push_back(ind); ind=v[i][ind]-1; } reverse(all(ans)); cout<<mval.ff<<"\n"; for(ll i:ans) { cout<<i+1<<" "; } nl; cerr<<"\nTime elapsed: "<<1000*clock()/CLOCKS_PER_SEC<<"ms\n"; return 0; }
Compilation message
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 108 == 108 |
2 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 999 == 999 |
3 | Correct | 1 ms | 212 KB | contestant found the optimal answer: 0 == 0 |
4 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 1542524 == 1542524 |
5 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 4500000000 == 4500000000 |
6 | Correct | 0 ms | 212 KB | contestant found the optimal answer: 1 == 1 |
7 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 1 == 1 |
8 | Correct | 0 ms | 212 KB | contestant found the optimal answer: 1 == 1 |
9 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 100400096 == 100400096 |
10 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 900320000 == 900320000 |
11 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 3698080248 == 3698080248 |
12 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 3200320000 == 3200320000 |
13 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 140072 == 140072 |
14 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 376041456 == 376041456 |
15 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 805 == 805 |
16 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 900189994 == 900189994 |
17 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 999919994 == 999919994 |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 1093956 == 1093956 |
2 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 302460000 == 302460000 |
3 | Correct | 0 ms | 468 KB | contestant found the optimal answer: 122453454361 == 122453454361 |
4 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 93663683509 == 93663683509 |
5 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 1005304678 == 1005304678 |
6 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 933702 == 933702 |
7 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 25082842857 == 25082842857 |
8 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 687136 == 687136 |
9 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 27295930079 == 27295930079 |
10 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 29000419931 == 29000419931 |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 610590000 == 610590000 |
2 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 311760000 == 311760000 |
3 | Correct | 1 ms | 1364 KB | contestant found the optimal answer: 1989216017013 == 1989216017013 |
4 | Correct | 0 ms | 340 KB | contestant found the optimal answer: 1499437552673 == 1499437552673 |
5 | Correct | 1 ms | 980 KB | contestant found the optimal answer: 1019625819 == 1019625819 |
6 | Correct | 1 ms | 1236 KB | contestant found the optimal answer: 107630884 == 107630884 |
7 | Correct | 2 ms | 1364 KB | contestant found the optimal answer: 475357671774 == 475357671774 |
8 | Correct | 1 ms | 468 KB | contestant found the optimal answer: 193556962 == 193556962 |
9 | Correct | 1 ms | 468 KB | contestant found the optimal answer: 482389919803 == 482389919803 |
10 | Correct | 1 ms | 596 KB | contestant found the optimal answer: 490686959791 == 490686959791 |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 21503404 == 21503404 |
2 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 140412195 == 140412195 |
3 | Correct | 8 ms | 2004 KB | contestant found the optimal answer: 49729674225461 == 49729674225461 |
4 | Correct | 1 ms | 340 KB | contestant found the optimal answer: 37485571387523 == 37485571387523 |
5 | Correct | 7 ms | 2004 KB | contestant found the optimal answer: 679388326 == 679388326 |
6 | Correct | 6 ms | 1796 KB | contestant found the optimal answer: 4699030287 == 4699030287 |
7 | Correct | 9 ms | 2012 KB | contestant found the optimal answer: 12418819758185 == 12418819758185 |
8 | Correct | 6 ms | 2036 KB | contestant found the optimal answer: 31093317350 == 31093317350 |
9 | Correct | 2 ms | 596 KB | contestant found the optimal answer: 12194625429236 == 12194625429236 |
10 | Correct | 3 ms | 980 KB | contestant found the optimal answer: 12345131038664 == 12345131038664 |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 2 ms | 852 KB | contestant found the optimal answer: 1818678304 == 1818678304 |
2 | Correct | 2 ms | 980 KB | contestant found the optimal answer: 1326260195 == 1326260195 |
3 | Correct | 70 ms | 9336 KB | contestant found the optimal answer: 4973126687469639 == 4973126687469639 |
4 | Correct | 2 ms | 852 KB | contestant found the optimal answer: 3748491676694116 == 3748491676694116 |
5 | Correct | 54 ms | 5972 KB | contestant found the optimal answer: 1085432199 == 1085432199 |
6 | Correct | 58 ms | 6796 KB | contestant found the optimal answer: 514790755404 == 514790755404 |
7 | Correct | 66 ms | 7216 KB | contestant found the optimal answer: 1256105310476641 == 1256105310476641 |
8 | Correct | 48 ms | 5964 KB | contestant found the optimal answer: 3099592898816 == 3099592898816 |
9 | Correct | 55 ms | 6812 KB | contestant found the optimal answer: 1241131419367412 == 1241131419367412 |
10 | Correct | 71 ms | 8452 KB | contestant found the optimal answer: 1243084101967798 == 1243084101967798 |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
1 | Correct | 17 ms | 5236 KB | contestant found the optimal answer: 19795776960 == 19795776960 |
2 | Correct | 20 ms | 5720 KB | contestant found the optimal answer: 19874432173 == 19874432173 |
3 | Correct | 684 ms | 81752 KB | contestant found the optimal answer: 497313449256899208 == 497313449256899208 |
4 | Correct | 18 ms | 5844 KB | contestant found the optimal answer: 374850090734572421 == 374850090734572421 |
5 | Correct | 865 ms | 81852 KB | contestant found the optimal answer: 36183271951 == 36183271951 |
6 | Correct | 546 ms | 58452 KB | contestant found the optimal answer: 51629847150471 == 51629847150471 |
7 | Correct | 646 ms | 62280 KB | contestant found the optimal answer: 124074747024496432 == 124074747024496432 |
8 | Correct | 468 ms | 51424 KB | contestant found the optimal answer: 309959349080800 == 309959349080800 |
9 | Correct | 513 ms | 58320 KB | contestant found the optimal answer: 124113525649823701 == 124113525649823701 |
10 | Correct | 669 ms | 73952 KB | contestant found the optimal answer: 124309619349406845 == 124309619349406845 |