Submission #791036

#	Time	Username	Problem	Language	Result	Execution time	Memory
791036		MarwenElarbi	Palembang Bridges (APIO15_bridge)	C++17	22 / 100	271 ms	24444 KiB

bridge

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#define vi vector<int>
#define ve vector
#define ll long long
#define vl vector<ll>
#define vll vector<pair<ll,ll>>
#define onbit __builtin_popcount
#define ii pair<int,int>
#define vvi vector<vi>
#define vii vector<ii>
#define gii greater<ii>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define INF 1e18
#define eps 1e-3
#define eps1 1e-2
#define optimise ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
#define MAX_A 1e5+5
using namespace std;
using namespace __gnu_pbds;
template <class T>
using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
const ll MOD = 1e9+7;
const int nax = 5e5+5;
const int MAX_VAL = 1e6+1;
double PI=3.14159265359;
int arx[8]={1,1,0,-1,-1,-1, 0, 1};
int ary[8]={0,1,1, 1, 0,-1,-1,-1};
void setIO(string s) {
    freopen((s + ".in").c_str(), "r", stdin);
    freopen((s + ".out").c_str(), "w", stdout);
}
int main(){
    //setIO("redistricting");
    int k,n;
    cin>>k>>n;
    long long ans=0;
    map<int,int> nabba;
    Tree<pair<int,int>> tab;
    for (int i = 0; i < n; ++i)
    {
        char a,b;
        int x,y;
        cin>>a>>x>>b>>y;
        if (a==b) ans+=abs(x-y);
        else{
            nabba[x]++;
            auto cur=nabba[x];
            tab.insert({x,nabba[x]});
            nabba[y]++;
            tab.insert({y,nabba[y]});
        }
    }
    int m=tab.size();
    if (k==1&&m!=0){
        int h=(*tab.find_by_order((m-1)/2)).first;
        //cout <<h<<endl;
        for (auto u:tab)
        {
            //cout <<u.fi<<" ";
            ans+=abs(h-u.fi);
        }
        cout <<ans+m/2<<endl;
    }else{
        cout <<ans<<endl;
    }
}

Compilation message (stderr)

bridge.cpp: In function 'int main()':
bridge.cpp:51:18: warning: unused variable 'cur' [-Wunused-variable]
   51 |             auto cur=nabba[x];
      |                  ^~~
bridge.cpp: In function 'void setIO(std::string)':
bridge.cpp:33:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |     freopen((s + ".in").c_str(), "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
bridge.cpp:34:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   34 |     freopen((s + ".out").c_str(), "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

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