Submission #473599

#	Time	Username	Problem	Language	Result	Execution time	Memory
473599		NicolaAbusaad2014	Split the sequence (APIO14_sequence)	C++14	50 / 100	1226 ms	131076 KiB

sequence

    /**
     * Prof.Nicola
    **/
    #include <bits/stdc++.h>
     
    using namespace std;
    #define endl "\n"
    #define mp make_pair
    #define F first
    #define S second
    #define REP(i,l,r) for(long long i=(l);i<(r);i++)
    #define PER(i,l,r) for(long long i=(r)-1;i>=(l);i--)
    const long dx[4]={1,0,-1,0},dz[4]={0,1,0,-1};
    const long double pi=3.14159265359;
    const long long mod=1e9+7;
    long long p(long long x){while(x&(x-1)){x=x&(x-1);}return x;}
    long long squared(long long x){return (x*x)%mod;}
    long long power(long long x,long long p){if(p==0){return 1;}if(p%2==1){return (power(x,p-1)*x)%mod;}return squared(power(x,p/2));}
    long long inv(long long x){return power(x,mod-2);}
    template<class T>void re(T&x){cin>>x;}
    template<class T1,class T2> void re(pair<T1,T2>&x){re(x.first);re(x.second);}
    template<class T>void re(vector<T>&x){for(long i=0;i<x.size();i++){re(x[i]);}}
    template<class T>void re(deque<T>&x){for(long i=0;i<x.size();i++){re(x[i]);}}
    template<class T>void out(T x){cout<<x<<" ";}
    template<class T1,class T2>void out(pair<T1,T2>x){out(x.first);out(x.second);cout<<endl;}
    template<class T>void out(vector<T>x,long l=0,long r=0){if(!r){r=x.size();}for(long i=l;i<r;i++){out(x[i]);}cout<<endl;}
    template<class T>void out(deque<T>x,long l=0,long r=0){if(!r){r=x.size();}for(long i=l;i<r;i++){out(x[i]);}cout<<endl;}
    template<class T>void out(set<T>x){while(!x.empty()){out(*x.begin());x.erase(*x.begin());}cout<<endl;}
    template<class T1,class T2>void out(map<T1,T2>x){while(!x.empty()){out(*x.begin());x.erase(x.begin()->first);}cout<<endl;}
    template<class T>void out(queue<T>x){while(!x.empty()){out(x.front());x.pop();}cout<<endl;}
    template<class T>void out(priority_queue<T>x){while(!x.empty()){out(x.top());x.pop();}cout<<endl;}
    template<class T>void out(stack<T>x){while(!x.empty()){out(x.top());x.pop();}cout<<endl;}
    template<class T>T cross(complex<T>x,complex<T>z){return (conj(x)*z).imag();}
    template<class T>T dot(complex<T>x,complex<T>z){return (conj(x)*z).real();}
    set<long long>::iterator T;
    long long vLE(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>x){return -1;}long long z=p(r-l);while(z){if(l+z<=r&&VT[l+z]<=x){l+=z;}z/=2;}return l;}
    long long vL(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>=x){return -1;}long long z=p(r-l);while(z){if(l+z<=r&&VT[l+z]<x){l+=z;}z/=2;}return l;}
    long long vGE(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>=x){return l;}l=vL(x,VT,l,r);if(l==r){return -1;}return l+1;}
    long long vG(long long x,vector<long long>&VT,long l=0,long r=-1){if(r==-1){r=VT.size()-1;}if(VT[l]>x){return l;}l=vLE(x,VT,l,r);if(l==r){return -1;}return l+1;}
    long long sLE(long long x,set<long long>&ST){if(ST.count(x)){return x;}if(*ST.begin()>x){return -1;}ST.insert(x);T=ST.find(x);T--;ST.erase(x);return *T;}
    long long sL(long long x,set<long long>&ST){if(*ST.begin()>=x){return -1;}if(ST.count(x)){T=ST.find(x);T--;return *T;}ST.insert(x);T=ST.find(x);T--;ST.erase(x);return *T;}
    long long sGE(long long x,set<long long>&ST){if(ST.count(x)){return x;}if(*ST.rbegin()<x){return -1;}ST.insert(x);T=ST.find(x);T++;ST.erase(x);return *T;}
    long long sG(long long x,set<long long>&ST){if(*ST.rbegin()<=x){return -1;}if(ST.count(x)){T=ST.find(x);T++;return *T;}ST.insert(x);T=ST.find(x);T++;ST.erase(x);return *T;}
    typedef long long ll;
    struct Line {
    	mutable ll k, m, p ,idx;
    	bool operator<(const Line& o) const { return k < o.k; }
    	bool operator<(ll x) const { return p < x; }
    };
     
    struct LineContainer : multiset<Line, less<>> {
    	static const ll inf = LLONG_MAX;
    	ll div(ll a, ll b) {
    		return a / b - ((a ^ b) < 0 && a % b); }
    	bool isect(iterator x, iterator y) {
    		if (y == end()) return x->p = inf, 0;
    		if (x->k == y->k) x->p = x->m > y->m ? inf : -inf;
    		else x->p = div(y->m - x->m, x->k - y->k);
    		return x->p >= y->p;
    	}
    	void add(ll k, ll m, ll idx) {
    		auto z = insert({k, m, 0, idx}), y = z++, x = y;
    		while (isect(y, z)) z = erase(z);
    		if (x != begin() && isect(--x, y)) isect(x, y = erase(y));
    		while ((y = x) != begin() && (--x)->p >= y->p)
    			isect(x, erase(y));
    	}
    	pair<ll,int> query(ll x) {
    		assert(!empty());
    		auto l = *lower_bound(x);
    		return mp(l.k * x + l.m,l.idx);
    	}
    };
    vector<long long>pre;
    long long f(int l,int r)
    {
        if(l==0){
            return pre[r];
        }
        return pre[r]-pre[l-1];
    }
    int main()
    {
        ios_base::sync_with_stdio(false);
        cin.tie(NULL);
        long long n,k,z,mx=-1,x;
        cin>>n>>k;
        vector<long long>v(n);
        pre.resize(n);
        re(v);
        pre[0]=v[0];
        REP(i,1,n){
            pre[i]=pre[i-1]+v[i];
        }
        v.clear();
        vector<LineContainer>sol(k+1);
        vector<vector<int> >from(n,vector<int>(k+1));
        pair<long long,int>p;
        sol[0].add(0,0,0);
        REP(i,1,n){
            PER(j,1,k+1){
                if(j>i){
                    continue;
                }
                p=sol[j-1].query(f(i,n-1));
                from[i][j]=p.second;
                z=p.first+(f(0,i-1)*f(i,n-1));
                sol[j].add(-f(0,i-1),z,i);
                if(j==k){
                    if(z>mx){
                        mx=z;
                        x=i;
                    }
                }
            }
        }
        cout<<mx<<endl;
        REP(i,0,k){
            cout<<x<<" ";
            x=from[x][k-i];
        }
        return 0;
    }

Compilation message (stderr)

sequence.cpp: In instantiation of 'void re(std::vector<_Tp>&) [with T = long long int]':
sequence.cpp:90:13:   required from here
sequence.cpp:22:57: warning: comparison of integer expressions of different signedness: 'long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   22 |     template<class T>void re(vector<T>&x){for(long i=0;i<x.size();i++){re(x[i]);}}
      |                                                        ~^~~~~~~~~
sequence.cpp: In function 'int main()':
sequence.cpp:120:21: warning: 'x' may be used uninitialized in this function [-Wmaybe-uninitialized]
  120 |             x=from[x][k-i];
      |                     ^

• Subtask 1: 11 / 11
• Subtask 2: 11 / 11
• Subtask 3: 11 / 11
• Subtask 4: 17 / 17
• Subtask 5: 0 / 21
• Subtask 6: 0 / 29