| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 784753 | YassirSalama | Shortcut (IOI16_shortcut) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "shortcut.h"
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll MAXN=5100;
const ll INF=1e9+10;
#define dbg(x) cout<<" [ "<<#x<<" ] : "<<x<<endl;
#define F first
#define S second
#define all(v) v.begin(),v.end()
long long find_shortcut(int n, vector<int> l, vector<int> d, int c)
{
if(n==2){
if(c<l[0])
return d[0]+c+d[1];
return d[0]+l[0]+d[1];
}
vector<long long> pref(MAXN+1,0);
for(long long i=0;i<n-1;i++){
pref[i+1]=pref[i]+l[i];
}
vector<long long> left(MAXN+1,0);
vector<long long> right(MAXN+1,0);
//the idea is to use pref[i-1]+l[i] or d[i]
left[0]=(ll)d[0];
long long first_node=0;
for(long long i=1;i<n;i++){
if(d[i]>left[i-1]+l[i-1]) first_node=i;
left[i]=max(left[i-1]+l[i-1],(ll)d[i]);
}
long long last_node=n-1;
right[n-1]=d[n-1];
for(long long i=n-2;i>=0;i--){
if(d[i]>right[i+1]+l[i]) last_node=i;
right[i]=max(right[i+1]+l[i],(ll)d[i]);
}
vector<long long> nodes;//nodes in the diameter
// cout<<first_node<<" "<<last_node<<endl;
long long diameter=left[first_node]+right[first_node];
// cout<<diameter<<endl;
for(long long i=first_node;i<=last_node;i++){
for(long long j=i+1;j<=last_node;j++){
long long express=left[i]+c+right[j];
long long newdiameter=0;
for(long long k=0;k<=j;k++){
newdiameter=max(newdiameter,(ll)abs(pref[j]-pref[k])+right[j]);
}
for(long long k=n-1;k>=i;k--){
newdiameter=max(newdiameter,(ll)abs(pref[k]-pref[i])+left[i]);
}
// if(i==0&&j==7){
// cout<<pref[j]-
// }
// cout<<newdiameter<<" "<<express<<" "<<i<<" "<<j<<endl;
// if(newdiameter>express) continue;
diameter=min({max(newdiameter,(ll)express),(ll)diameter});
}
}
// cout<<c<<endl;
return diameter;
}
int main()
{
int n, c;
assert(2 == scanf("%d%d", &n, &c));
std::vector<int> l(n - 1);
std::vector<int> d(n);
for (int i = 0; i < n - 1; i++)
assert(1 == scanf("%d", &l[i]));
for (int i = 0; i < n; i++)
assert(1 == scanf("%d", &d[i]));
long long t = find_shortcut(n, l, d, c);
printf("%lld\n", t);
return 0;
}