Submission #74838

#	Time	Username	Problem	Language	Result	Execution time	Memory
74838		cubecube1000	도로 건설 사업 (JOI13_construction)	C++14	10 / 100	1520 ms	263168 KiB

construction

#include<bits/stdc++.h>
#define x first
#define y second
using namespace std;
typedef unsigned long long ll;
typedef pair<int,int> pii;
typedef pair<ll, ll> pll;
const ll MAX=1<<18;
const ll INFLL=0x3f3f3f3f3f3f3f;
struct vill{
    ll x,y,idx;
};
vill v[MAX];
struct edge{
    ll x,y,d;
    bool operator<(const edge &a)const{
        return d<a.d;
    }
};
struct area{ll x1,y1,x2,y2;};
area h[MAX];
vector<ll> ixcp,iycp,ex[MAX],ey[MAX],val,val2;
vector<edge> mst;
vector<pll> conn[MAX];
map<ll,ll> xcp,ycp;
multiset<ll> pos;
vector<pll> ch;
vector<ll> ch_val;
ll chk[MAX],n,m,c,par[MAX],cum;
ll fnd(ll x){
    if(par[x]!=x) return par[x]=fnd(par[x]);
    return x;
}
void uni(ll x,ll y){
    x=fnd(x), y=fnd(y);
    par[x]=y;
}
bool cmpx(vill a,vill b){
    if(a.x!=b.x) return a.x<b.x;
    return a.y<b.y;
}
bool cmpy(vill a,vill b){
    if(a.y!=b.y) return a.y<b.y;
    return a.x<b.x;
}
bool cmpidx(vill a,vill b){
    return a.idx<b.idx;
}
ll exist(ll x,ll y){
    set<ll>::iterator it=pos.lower_bound(x);
    if(it==pos.end()) return false;
    if(*it<=y) return true;
    return false;
}
int main(){
    scanf("%lld%lld%lld",&n,&m,&c);
    for(ll i=0;i<n;i++){scanf("%lld%lld",&v[i].x,&v[i].y),v[i].idx=i;}
    for(ll i=0;i<m;i++){scanf("%lld%lld%lld%lld",&h[i].x1,&h[i].y1,&h[i].x2,&h[i].y2),h[i].x2++,h[i].y2++;}
    for(ll i=0;i<n;i++) {
        if(xcp.find(h[i].x1)==xcp.end()) ixcp.push_back(h[i].x1), xcp[h[i].x1]=0;
        if(xcp.find(h[i].x2)==xcp.end()) ixcp.push_back(h[i].x2), xcp[h[i].x2]=0;
        if(ycp.find(h[i].y1)==ycp.end()) iycp.push_back(h[i].y1), ycp[h[i].y1]=0;
        if(ycp.find(h[i].y2)==ycp.end()) iycp.push_back(h[i].y2), ycp[h[i].y2]=0;
    }
    sort(ixcp.begin(),ixcp.end());
    sort(iycp.begin(),iycp.end());
    for(ll i=0;i<ixcp.size();i++) xcp[ixcp[i]]=i;
    for(ll i=0;i<iycp.size();i++) ycp[iycp[i]]=i;
    for(ll i=0;i<m;i++){
        ex[xcp[h[i].x1]].push_back(i), ey[ycp[h[i].y1]].push_back(i);
        ex[xcp[h[i].x2]].push_back(i), ey[ycp[h[i].y2]].push_back(i);
    }
    sort(v,v+n,cmpx);
    for(ll i=1,pt=0;i<n;i++){
        if(v[i-1].x==v[i].x){
            for(;ixcp[pt]<=v[i].x&&pt<ixcp.size();pt++){
                for(ll j=0;j<ex[pt].size();j++) {
                    if(chk[ex[pt][j]]) pos.erase(h[ex[pt][j]].y1),pos.erase(h[ex[pt][j]].y2-1);
                    else pos.insert(h[ex[pt][j]].y1),pos.insert(h[ex[pt][j]].y2-1);
                    chk[ex[pt][j]]^=1;
                }
            }
            if(!exist(v[i-1].y,v[i].y)){
                ll dist=v[i].y-v[i-1].y;
                conn[v[i].idx].push_back(make_pair(v[i-1].idx,dist));
                conn[v[i-1].idx].push_back(make_pair(v[i].idx,dist));
            }
        }
    }
    pos.clear();
    for(ll i=0;i<m;i++) chk[i]=0;
    sort(v,v+n,cmpy);
    for(ll i=1,pt=0;i<n;i++){
        if(v[i-1].y==v[i].y){
            for(;iycp[pt]<=v[i].y&&pt<iycp.size();pt++){
                for(ll j=0;j<ey[pt].size();j++) {
                    if(chk[ey[pt][j]]) pos.erase(h[ey[pt][j]].x1),pos.erase(h[ey[pt][j]].x2-1);
                    else pos.insert(h[ey[pt][j]].x1),pos.insert(h[ey[pt][j]].x2-1);
                    chk[ey[pt][j]]^=1;
                }
            }
            //printf("%lld%lld%lld\n",pos.size(),v[i].y,iycp.size());
            if(!exist(v[i-1].x,v[i].x)){
                ll dist=v[i].x-v[i-1].x;
                conn[v[i].idx].push_back(make_pair(v[i-1].idx,dist));
                conn[v[i-1].idx].push_back(make_pair(v[i].idx,dist));
            }
        }
    }
    for(ll i=0;i<n;i++){
        for(ll j=0;j<conn[i].size();j++) {
            if(i<conn[i][j].x) mst.push_back({i,conn[i][j].x,conn[i][j].y});
            //printf("%lld%lld%lld\n",i,conn[i][j].x,conn[i][j].y);
        }
    }

    for(ll i=0;i<n;i++) par[i]=i;
    sort(mst.begin(),mst.end());
    val.push_back(0),val2.push_back(0);
    for(ll i=0;i<mst.size();i++){
        if(fnd(mst[i].x)!=fnd(mst[i].y)) cum+=mst[i].d, val.push_back(cum), val2.push_back(mst[i].d), uni(mst[i].x,mst[i].y);
    }
    //for(ll i=0;i<val.size();i++) printf("%lld%lld\n",n-i,val[i]);
    vector<ll>::iterator it;
    for(ll i=0;i<c;i++){
        ll t1,t2;
        scanf("%lld%lld",&t1,&t2);
        if(n-val.size()>=t2){
            printf("-1\n");
        }
        else{
            it=lower_bound(val2.begin(),val2.end(),t1);
            ll idx=max((ll)(it-val2.begin()-1),n-t2);
            printf("%lld\n",val[idx]+t1*(n-idx));
        }
    }
}

Compilation message (stderr)

construction.cpp: In function 'int main()':
construction.cpp:56:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%lld%lld%lld",&n,&m,&c);
     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
construction.cpp:57:58: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     for(ll i=0;i<n;i++){scanf("%lld%lld",&v[i].x,&v[i].y),v[i].idx=i;}
                         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~
construction.cpp:58:96: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     for(ll i=0;i<m;i++){scanf("%lld%lld%lld%lld",&h[i].x1,&h[i].y1,&h[i].x2,&h[i].y2),h[i].x2++,h[i].y2++;}
                         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~
construction.cpp:127:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld%lld",&t1,&t2);
         ~~~~~^~~~~~~~~~~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 30
• Subtask 3: 0 / 30
• Subtask 4: 0 / 30