Submission #1102789

#	Time	Username	Problem	Language	Result	Execution time	Memory
1102789		PedroBigMan	Nile (IOI24_nile)	C++17	67 / 100	2060 ms	7096 KiB

nile

#include "nile.h"
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#pragma GCC optimize("Ofast")
#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
#include <string>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <queue>
#include <deque>
#include <list>
#include <iomanip>
#include <stdlib.h>
#include <time.h>
#include <cstring>
using namespace std;
typedef long long int ll;
typedef unsigned long long int ull;
typedef long double ld;
#define REP(i,a,b) for(ll i=(ll) a; i<(ll) b; i++)
#define pb push_back
#define mp make_pair
#define pl pair<ll,ll>
#define ff first
#define ss second
#define whole(x) x.begin(),x.end()
#define DEBUG(i) cout<<"Pedro "<<i<<endl
#define INF 1000000000000000000LL
#define EPS ((ld)0.00000000001)
#define pi ((ld)3.141592653589793)
#define VV(vvvv,NNNN,xxxx); REP(iiiii,0,NNNN) {vvvv.pb(xxxx);}
ll mod=1000000007;

template<class A=ll> 
void Out(vector<A> a) {REP(i,0,a.size()) {cout<<a[i]<<" ";} cout<<endl;}

template<class A=ll>
void In(vector<A> &a, ll N) {A cur; REP(i,0,N) {cin>>cur; a.pb(cur);}} 


vector<long long> calculate_costs(vector<int> W, vector<int> A, vector<int> B, vector<int> E) 
{
  ll Q = E.size(); ll N = A.size();
  vector<pl> a; 
  ll basis = 0LL; REP(i,0,N) {basis+=A[i]; a.pb({W[i],A[i]-B[i]});}
  sort(whole(a));
  ll D; ll prev;
  vector<ll> ans(Q,0);
  REP(q,0,Q)
  {
    D = E[q];
    vector<ll> dp(N,0);
    REP(i,0,N)
    {
      if(i>=1) {dp[i]=max(dp[i],dp[i-1]);}
      if(i>=1 && a[i].ff-a[i-1].ff<=D) {prev=0; if(i>1) {prev=dp[i-2];} dp[i]=max(dp[i],prev+a[i-1].ss+a[i].ss);}
      if(i>=2 && a[i].ff-a[i-2].ff<=D) {prev=0; if(i>2) {prev=dp[i-3];} dp[i]=max(dp[i],prev+a[i-2].ss+a[i].ss);}
    }
    ans[q]=basis-dp[N-1];
  }
  return ans;
}

Compilation message (stderr)

nile.cpp:2: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    2 | #pragma GCC optimization ("O3")
      | 
nile.cpp:3: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    3 | #pragma GCC optimization ("unroll-loops")
      |

• Subtask 0: 0 / 0
• Subtask 1: 6 / 6
• Subtask 2: 13 / 13
• Subtask 3: 17 / 17
• Subtask 4: 11 / 11
• Subtask 5: 20 / 20
• Subtask 6: 0 / 15
• Subtask 7: 0 / 18