Submission #1219430

#	Time	Username	Problem	Language	Result	Execution time	Memory
1219430		YassirSalama	Palembang Bridges (APIO15_bridge)	C++20	100 / 100	157 ms	22356 KiB

bridge

#include <bits/stdc++.h>
using namespace std;
#define endl "\n"
#define int long long
using ull=unsigned long long;
using pii=pair<int,int>;
const int dx[4] = {1,0,-1,0}, dy[4] = {0,1,0,-1};
#define OVL(x,s) for(auto y:x) cout<<y<<s; cout<<"\n";
template <typename T> istream& operator>>(istream& is, vector<T> &a) {
    copy_n(istream_iterator<T>(is), a.size(), a.begin()); return is;}
#ifdef IOI
template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << '(' << p.first << ", " << p.second << ')'; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { os << '{'; string sep; for (const T &x : v) os << sep << x, sep = ", "; return os << '}'; }

void dbg_out() { cout << endl; }
template<typename Head, typename... Tail> void dbg_out(Head H, Tail... T) { cout << ' ' << H; dbg_out(T...); }
#define dbg(...) cout << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__);
#else
#define dbg(...) 1337;
#endif
#define pb push_back
#define F first
#define S second
#define all(v) v.begin(),v.end()
const int inf = (1LL<<60);
const int maxn = 2e5+100;
struct median{
    multiset<int> left,right;
    //at everytime |left|==|right| or |left|==|right|+1
    int sl = 0;
    int sr = 0;
    void add(int x){
        if(left.empty()||x<=*left.rbegin()){
            left.insert(x);
            sl+=x;
        }else{
            right.insert(x);
            sr+=x;
        }
        while(left.size()<right.size()){
            auto it = *right.begin();
            right.erase(right.find((it)));
            sr-=it;
            sl+=it;
            left.insert(it);
        }
        while(left.size()>right.size()+1){
            auto it = *left.rbegin();
            sl-=it;
            sr+=it;
            left.erase(left.find(it));
            right.insert(it);
        }
    }
    int med(){
        return *left.rbegin();
    }
    int cost(){
        int cntl = left.size();
        int cntr = right.size();
        int m = med();
        return m*cntl-sl+sr-m*cntr;
    }
};

signed main() {
	ios_base::sync_with_stdio(0);cin.tie(0);
	int n,k;
	cin>>k>>n;
    if(k==1){
        map<int,int> mp;
        int t = 0;
        int suff = 0;
        int cnt = 0;
        for(int i = 0;i<n;i++){
            char c,cc;
            int a,b;
            cin>>c>>a>>cc>>b;
            if(c==cc){
                t+=abs(b-a);
            }else{
                mp[a]++;
                mp[b]++;
                cnt+=2;
                suff+=b;
                suff+=a;
            }
        }
        int ans = 1e18;
        int pref = 0;
        int cnl = 0;
        for(auto x : mp){
            int cost = cnl*x.F-pref;
            int c2 = suff-cnt*x.F;
            ans = min(ans,cost+c2+(cnl+cnt)/2); 
            cnt-= x.S;
            suff-= x.S * x.F;
            cnl+= x.S;
            pref+= x.S * x.F;
        }
        if(mp.size()==0) ans = 0;
        cout<<ans+t<<endl;
        return  0;
    }
    vector<pair<int,int>> v;
    int s = 0;
    for(int i = 0;i<n;i++){
        char c,cc;
        int a,b;
        cin>>c>>a;
        cin>>cc>>b;
        if(c==cc){
            s+=abs(a-b);
        }else{
            v.pb({a,b});
        }
    }
    sort(all(v),[&](auto a,auto b){
        return a.F+a.S<b.F+b.S;
    });
    if(v.empty()){
        cout<<s<<endl;
        return 0;
    }
    int pref[(int)v.size()];
    memset(pref,0,sizeof(pref));
    median m;
    for(int i = 0;i<v.size();i++){
        m.add(v[i].F);
        m.add(v[i].S);
        pref[i] = m.cost();
    }
    median mm;
    int suff[(int)v.size()];
    for(int i = v.size()-1;i>=0;i--){
        mm.add(v[i].F);
        mm.add(v[i].S);
        suff[i] = mm.cost();
    }
    int ans = 1e18;
    for(int i = 0;i+1<v.size();i++){
        ans = min(ans,pref[i]+suff[i+1]);
    }
    ans = min(ans,suff[0]);
    dbg(ans)
    ans+=v.size();
    cout<<ans+s<<endl;
}

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