이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "wiring.h"
#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define INF 10000000
#define MOD 1000000007
#define MID ((l+r)/2)
#define HASHMOD 2305843009213693951ll
#define ll long long
#define ull unsigned long long
#define F first
#define S second
typedef pair<ll, ll> ii;
typedef pair<ii, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef map<int, int> mii;
#define EPS 1e-6
#define FOR(i,n) for(int i=0;i<((int)(n));i++)
#define FORi(i,a,b) for(int i=((int)(a));i<((int)(b));i++)
#define FOA(v, a) for(auto v : a)
map<int, int> used;
ll dp[3000000];
vector<ii> all;
int nxt(int x){
int orx=x;
while(x<(int)all.size() && all[x].S==all[orx].S) x++;
return x;
}
ll ps[3000000];
ll sum(int l, int r){ // get subarray sum, [l, r)
assert(ps[r]>=ps[l]);
return ps[r] - ps[l];
}
long long min_total_length(std::vector<int> r, std::vector<int> b) {
int n = r.size();
int m = b.size();
FOA(v, r) all.push_back({v, 0});
FOA(v, b) all.push_back({v, 1});
sort(all.begin(), all.end());
ps[0] = all[0].F;
FOR(i,n+m-1){
ps[i+1] = ps[i] + all[i+1].F;
}
#ifdef LOCAL
FOR(i,n+m) cout<<all[i].F<<" ";
cout<<endl;
FOR(i,n+m) cout<<all[i].S<<" ";
cout<<endl;
#endif
dp[0] = 0;
FOR(i,n+m) dp[i+1] = HASHMOD;
int prev=-1, next=nxt(0);
int next2=nxt(next);
FOR(i,n+m){
if(i==next){
prev = i-1;
next = next2;
next2 = nxt(next2);
}
#ifdef LOCAL
cout<<i<<" "<<prev<<" "<<next<<" "<<next2<<endl;
#endif
if(prev>=0) dp[i+1] = min(dp[i+1], dp[i] + all[i].F - all[prev].F);
if(next<(int)all.size()) dp[i+1] = min(dp[i+1], dp[i] + all[next].F - all[i].F);
if(next2-next >=next-i){
dp[next + (next - i)] = min(dp[next + (next-i)], dp[i] + sum(next, next+(next-i)) - sum(i, next));
}
}
#ifdef LOCAL
FOR(i,n+m) cout<<i<<" ";
cout<<endl;
FOR(i,n+m) cout<<dp[i+1]<<" ";
cout<<endl;
#endif
return dp[n+m];
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |