oj.uz

Submission #242670

# Submission time Handle Problem Language Result Execution time Memory
242670 2020-06-28T22:55:52 Z AmineWeslati Feast (NOI19_feast) C++14
22 / 100
 1000 ms 8856 KB 
//Never stop trying
#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;

template <typename T>
using ordered_set=tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
//s.order_of_key(), *s.find_by_order()

using namespace std;
#define boost ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0)

typedef string str;
typedef long long ll;
#define int ll
typedef double db;
typedef long double ld;

typedef pair<int,int> pi;
#define fi first
#define se second

typedef vector<int> vi;
typedef vector<ld> vd;
typedef vector<str> vs;
typedef vector<pi> vpi;

#define pb push_back
#define eb emplace_back
#define pf push_front

#define lb lower_bound
#define ub upper_bound

#define sz(x) (int)x.size()
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
#define endl "\n"

const int MOD = 1e9+7; //998244353
const ll INF = 1e18;
const int nx[4]={0,0,1,-1}, ny[4]={1,-1,0,0}; //right left down up

int t[300001];
int pref[300001];
int memo[300][301];
int N,K;

int solve(int i, int r){
   if(!r) return 0;
   if(i==N) return -INF;
   int ans=0;
   for(int j=i; j<N; j++){
      for(int k=j+1; k<N; k++){
         ans=max(ans,solve(k,r-1)+pref[j]-pref[i-1]);
      }
   }
   return memo[i][r]=ans;
}

int32_t main(){
   boost;
   memset(memo,-1,sizeof(memo));
   cin>>N>>K;
   pref[0]=0;
   int cnt=0;
   for(int i=0; i<N; i++){
      cin>>t[i];
      pref[i+1]=t[i];
      if(i) pref[i+1]+=pref[i];
      if(t[i]<0) cnt++;
   }
   if(!cnt ||(cnt==1 && K>1)){
      int res=0;
      for(int i=0; i<N; i++) if(t[i]>0) res+=t[i];
      cout << res << endl;
   }
   else if(cnt==1){
      int s=0,ss=0,sss=0;
      int idx=0;
      while(t[idx]>=0) s+=t[idx];
      sss+=t[idx]; idx++;
      while(t[idx]>=0) ss+=t[idx];
      cout << max(max(s,ss),s+ss+sss) << endl;
   }
   else if(K==1){
      int s=0;
      int res=0;
      for(int i=0; i<N; i++){
         s+=t[i];
         res=max(res,s);
         if(s<=0) s=0;
      }
      cout << res << endl;
   }
   else if(N<=300){
      int res=0;
      for(int i=0; i<N; i++) res=max(res,solve(i,K));
      cout << res << endl;
   }


   return 0;
}

Subtask #1
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 45 ms 8440 KB Output is correct
2 Correct 51 ms 8568 KB Output is correct
3 Correct 50 ms 8568 KB Output is correct
4 Correct 46 ms 8572 KB Output is correct
5 Correct 46 ms 8576 KB Output is correct
6 Correct 47 ms 8568 KB Output is correct
7 Correct 48 ms 8440 KB Output is correct
8 Correct 54 ms 8572 KB Output is correct
9 Correct 44 ms 8440 KB Output is correct
10 Correct 45 ms 8440 KB Output is correct

Subtask #2
0 / 8.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1097 ms 6784 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #3
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 53 ms 8696 KB Output is correct
2 Correct 51 ms 8568 KB Output is correct
3 Correct 49 ms 8696 KB Output is correct
4 Correct 56 ms 8568 KB Output is correct
5 Correct 50 ms 8696 KB Output is correct
6 Correct 52 ms 8696 KB Output is correct
7 Correct 49 ms 8696 KB Output is correct
8 Correct 49 ms 8700 KB Output is correct
9 Correct 56 ms 8856 KB Output is correct
10 Correct 52 ms 8824 KB Output is correct

Subtask #4
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 1024 KB Output is correct
2 Execution timed out 1094 ms 1024 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #5
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 1024 KB Output is correct
2 Execution timed out 1094 ms 1024 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #6
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 1024 KB Output is correct
2 Execution timed out 1094 ms 1024 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #7
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 45 ms 8440 KB Output is correct
2 Correct 51 ms 8568 KB Output is correct
3 Correct 50 ms 8568 KB Output is correct
4 Correct 46 ms 8572 KB Output is correct
5 Correct 46 ms 8576 KB Output is correct
6 Correct 47 ms 8568 KB Output is correct
7 Correct 48 ms 8440 KB Output is correct
8 Correct 54 ms 8572 KB Output is correct
9 Correct 44 ms 8440 KB Output is correct
10 Correct 45 ms 8440 KB Output is correct
11 Execution timed out 1097 ms 6784 KB Time limit exceeded
12 Halted 0 ms 0 KB -