제출 #951179

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
951179		JakobZorz	Palembang Bridges (APIO15_bridge)	C++17	100 / 100	363 ms	25324 KiB

bridge

#include<iostream>
#include<vector>
#include<queue>
#include<stack>
#include<algorithm>
#include<limits.h>
#include<math.h>
#include<map>
#include<set>
#include<unordered_map>
#include<unordered_set>
#include<iomanip>
#include<cstring>
#include<random>
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
using namespace std;

vector<ll>calc(vector<pair<int,int>>arr){ // res[i] = how much would it cost if i setup 1 bridge for first i people
    set<int>ptss;
    for(auto[i,j]:arr){
        ptss.insert(i);
        ptss.insert(j);
    }
    vector<int>pts(ptss.begin(),ptss.end());
    vector<ll>res(arr.size()+1,1e18);
    multiset<int>points1,points2;
    int mpc=0; // mid points count
    int br=0;
    ll cr=0;
    for(int i=0;i<=(int)arr.size();i++){
        if(i){
            if(arr[i-1].first==pts[br])
                mpc++;
            else if(arr[i-1].first>pts[br])
                points2.insert(arr[i-1].first);
            else
                points1.insert(arr[i-1].first);
            if(arr[i-1].second==pts[br])
                mpc++;
            else if(arr[i-1].second>pts[br])
                points2.insert(arr[i-1].second);
            else
                points1.insert(arr[i-1].second);
            cr+=abs(arr[i-1].first-pts[br]);
            cr+=abs(arr[i-1].second-pts[br]);
        }
        while(points1.size()+mpc<points2.size()&&br<(int)pts.size()-1){
            cr-=1LL*(pts[br+1]-pts[br])*(points2.size()-points1.size()-mpc);
            while(mpc--)
                points1.insert(pts[br]);
            br++;
            mpc=(int)points2.count(pts[br]);
            points2.erase(pts[br]);
        }
        while(points1.size()>points2.size()+mpc&&br>0){
            cr-=1LL*(pts[br]-pts[br-1])*(points1.size()-points2.size()-mpc);
            while(mpc--)
                points2.insert(pts[br]);
            br--;
            mpc=(int)points1.count(pts[br]);
            points1.erase(pts[br]);
        }
        /*for(int br:pts){
            ll cr=0;
            for(int j=0;j<i;j++)
                cr+=abs(br-arr[j].first)+abs(br-arr[j].second);
            res[i]=min(res[i],cr);
        }
        cout<<res[i]<<" "<<cr<<"\n";*/
        res[i]=cr;
    }
    return res;
}

bool cmp(pair<int,int>a,pair<int,int>b){
    return a.first+a.second<b.first+b.second;
}

void solve(){
    int k,n;
    cin>>k>>n;
    vector<pair<int,int>>arr;
    ll res1=0;
    for(int i=0;i<n;i++){
        char c1,c2;
        int a1,a2;
        cin>>c1>>a1>>c2>>a2;
        if(c1==c2){
            res1+=abs(a1-a2);
        }else{
            res1++;
            if(a1>a2)
                swap(a1,a2);
            arr.emplace_back(a1,a2);
        }
    }
    
    sort(arr.begin(),arr.end(),cmp);
    
    vector<ll>r1=calc(arr);
    reverse(arr.begin(),arr.end());
    vector<ll>r2=calc(arr);
    
    ll res2=1e18;
    if(k==1){
        res2=r1.back();
    }else{ // k == 2
        for(int i=0;i<=(int)arr.size();i++)
            res2=min(res2,r1[i]+r2[(int)arr.size()-i]);
    }
    
    if(arr.size()==0)
        res2=0;
    
    cout<<res1+res2<<"\n";
}
 
int main(){
    ios::sync_with_stdio(false);cout.tie(NULL);cin.tie(NULL);
    //freopen("/Users/jakob/cp_testing/test.txt","r",stdin);//freopen("output.out","w",stdout);
    int t=1;//cin>>t;
    while(t--)solve();
    return 0;
}
 
 
/*

 
1 5
B 0 A 4
B 1 B 3
A 5 B 7
B 2 A 6
B 1 A 7
 
24

2 5
B 0 A 4
B 1 B 3
A 5 B 7
B 2 A 6
B 1 A 7
 
22


 */

• Subtask 1: 8 / 8
• Subtask 2: 14 / 14
• Subtask 3: 9 / 9
• Subtask 4: 32 / 32
• Subtask 5: 37 / 37