Submission #380520

#	Time	Username	Problem	Language	Result	Execution time	Memory
380520		jass921026	Palembang Bridges (APIO15_bridge)	C++14	22 / 100	59 ms	2300 KiB

bridge

#include<bits/stdc++.h>
using namespace std;
#define jizz ios_base::sync_with_stdio(false);cin.tie(NULL);
typedef long long ll;
typedef pair<int,int> pii;
#define F first
#define S second
#define ALL(x) (x).begin(),(x).end()
#define pb push_back
#define mkp make_pair
bool cmp(pii a, pii b){
    return a.F+a.S<b.F+b.S;
}
int abs(int x){
    return x>0?x:-x;
}
vector<pii> segs;
vector<int> pos, pos2;
int main(){ jizz
    int k, n;
    cin>>k>>n;
    ll tmp=0;
    for(int i=0;i<n;i++){
        char p, q;
        int s, t;
        cin>>p>>s>>q>>t;
        if(s>t) swap(s,t);
        tmp+=(t-s);
        if(p!=q) {
            tmp++;
            segs.pb(mkp(s,t));
            pos.pb(s);
            pos.pb(t);
        }
    }
    ll ans=tmp;
    if(pos.empty()){
        cout<<tmp<<"\n";
        return 0;
    }
    sort(ALL(pos));
    sort(ALL(segs),cmp);
    int sz=segs.size();
    int bp=pos[sz];
    for(int i=0;i<sz;i++){
        if(segs[i].S<bp) ans+=2*(bp-segs[i].S);
        else if(segs[i].F>bp) ans+=2*(segs[i].F-bp);
    }
    if(k==1){
        cout<<ans<<"\n";
        return 0;
    }
    ll ans2=ans;
    for(int i=0;i<sz;i++){
        ll res=tmp;
        pos.clear();
        pos2.clear();
        for(int j=0;j<sz;j++){
            if(j<=i){
                pos.pb(segs[i].F);
                pos.pb(segs[i].S);
            } else{
                pos2.pb(segs[i].F);
                pos2.pb(segs[i].S);
            }
        }
        sort(ALL(pos));
        sort(ALL(pos2));
        int bp1=pos[pos.size()/2], bp2=pos2[pos2.size()/2];
        for(int j=0;j<sz;j++){
            if(j<=i){
                if(segs[i].S<bp1) res+=2*(bp1-segs[i].S);
                else if(segs[i].F>bp1) res+=2*(segs[i].F-bp1);
            } else{
                if(segs[i].S<bp2) res+=2*(bp2-segs[i].S);
                else if(segs[i].F>bp2) res+=2*(segs[i].F-bp2);
            }
        }
        //cout<<i<<" "<<res<<"\n";
        ans2=min(ans2,res);
    }
    cout<<min(ans,ans2)<<"\n";
    return 0;
}
/*
1 5
B 0 A 4
B 1 B 3
A 5 B 7
B 2 A 6
B 1 A 7
2 5
B 0 A 4
B 1 B 3
A 5 B 7
B 2 A 6
B 1 A 7
*/

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