제출 #1097400

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1097400		alexdd	Homecoming (BOI18_homecoming)	C++17	62 / 100	1084 ms	188588 KiB

homecoming

#include <bits/stdc++.h>
#include "homecoming.h"
using namespace std;
const long long INF = 1e18;
long long dp[6000005][2];
long long a[6000005],initb[6000005],b[6000005];
long long prefb[6000005];
long long calc(int N, int K, int p)
{
    long long aux=0;
    for(int i=1;i<=N;i++)
    {
        if(i<=p)
        {
            b[i] = b[i+N] = 0;
            aux += initb[i];
        }
        else
            b[i] = b[i+N] = initb[i];
    }
    for(int i=1;i<=2*N;i++)
    {
        prefb[i] = prefb[i-1] + b[i];
    }
    for(int i=0;i<K;i++)
    {
        dp[i][0]=0;
        dp[i][1]=-INF;
    }
    for(int i=K;i<N+K;i++)
    {
        dp[i][1] = max(dp[i-1][1]+a[i]-b[i], dp[i-K][0]+a[i]-(prefb[i]-prefb[i-K]));
        dp[i][0] = max({dp[i-1][0], dp[i-1][1], dp[i][1]});
    }
    return max(dp[N+K-1][0], dp[N+K-1][1]) - aux;
}
long long int solve(int N, int K, int *copA, int *copB)
{
    for(int i=0;i<N;i++)
    {
        a[i+K] = copA[i];
        initb[i+1] = copB[i];
    }
    long long rez=0;
    for(int p=0;p<=K;p++)
        rez = max(rez, calc(N,K,p));
    return rez;
}

• Subtask 1: 13 / 13
• Subtask 2: 18 / 18
• Subtask 3: 31 / 31
• Subtask 4: 0 / 38