Submission #1059794

#	Time	Username	Problem	Language	Result	Execution time	Memory
1059794		tolbi	Closing Time (IOI23_closing)	C++17	9 / 100	1026 ms	11344 KiB

closing

#include "closing.h"

#include <bits/stdc++.h>
using namespace std;

int max_score(int N, int X, int Y, long long K,
  std::vector<int> U, std::vector<int> V, std::vector<int> W)
{
    vector<long long> dp1(N);
    vector<long long> dp2(N);
    vector<long long> dp3(N);
    long long cur = 0;
    for (int i = X-1; i >= 0; i--){
        cur+=W[i];
        dp1[i]=cur;
    }
    cur=0;
    for (int i = X+1; i < N; i++){
        cur+=W[i-1];
        dp1[i]=cur;
    }

    cur = 0;
    for (int i = Y-1; i >= 0; i--){
        cur+=W[i];
        dp2[i]=cur;
    }
    cur=0;
    for (int i = Y+1; i < N; i++){
        cur+=W[i-1];
        dp2[i]=cur;
    }
    for (int i = 0; i < N; i++){
        dp3[i]=max(dp1[i],dp2[i]);
        if (i) dp3[i]+=dp3[i-1];
    }
    int ans = 0;
    for (int l = 0; l <= Y; l++){
        for (int r = max(X,l); r < N; r++){
            long long cur = dp3[r];
            if (l) cur-=dp3[l-1];
            if (cur>K) break;
            int rbound = N-1;
            for (int j = r+1; j < N; j++){
                cur+=dp2[j];
            }
            for (int j = l-1; j >= 0; j--){
                cur+=dp1[j];
                while (cur>K){
                    if (rbound==r) break;
                    cur-=dp2[rbound];
                    rbound--;
                    if (rbound==r) break;
                }
                if (cur>K) break;
                if (cur<=K && rbound>=Y && j<=X){
                    ans=max(ans,rbound-l+1+r-j+1);
                }
            }
        }
    }
    vector<long long> pref = dp2;
    for (int i = 1; i < pref.size(); i++){
        pref[i]+=pref[i-1];
    }
    function<long long(int,int)> query = [&](int l, int r)->long long{
        if (l>r) return 0;
        long long hh = pref[r];
        if (l) hh-=pref[l-1];
        return hh;
    };
    for (int a = 0; a <= X; a++){
        for (int b = Y; b < N; b++){
            long long cur = 0;
            for (int i = a; i <= X; i++){
                cur+=dp1[i];
            }
            for (int i = Y; i <= b; i++){
                cur+=dp2[i];
            }
            if (cur>K) continue;
            for (int rv = a; rv < b; rv++){
                if (cur>K) break;
                int l = rv+1, r = b;
                while (l<r){
                    int mid = l+(r-l)/2;
                    if (query(mid,b-1)+cur<=K){
                        r=mid;
                    }
                    else l=mid+1;
                }
                ans=max(ans,rv-a+1+b-l+1);
                cur+=dp1[rv+1];
            }
        }
    }
    return ans;
}

Compilation message (stderr)

closing.cpp: In function 'int max_score(int, int, int, long long int, std::vector<int>, std::vector<int>, std::vector<int>)':
closing.cpp:63:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |     for (int i = 1; i < pref.size(); i++){
      |                     ~~^~~~~~~~~~~~~

• Subtask 0: 0 / 0
• Subtask 1: 0 / 8
• Subtask 2: 9 / 9
• Subtask 3: 0 / 12
• Subtask 4: 0 / 14
• Subtask 5: 0 / 9
• Subtask 6: 0 / 11
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 17