結果

問題 No.1012 荷物収集
ユーザー otamay6
提出日時 2019-11-27 20:03:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 335 ms / 2,000 ms
コード長 1,372 bytes
コンパイル時間 1,976 ms
コンパイル使用メモリ 185,084 KB
実行使用メモリ 17,076 KB
最終ジャッジ日時 2024-10-14 12:41:07
合計ジャッジ時間 12,051 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<unistd.h>
#define REP(i,n) for(int i=0,i##_len=(n);i<i##_len;++i)
#define rep(i,a,b) for(int i=int(a);i<int(b);++i)
#define All(x) (x).begin(),(x).end()
using namespace std;
typedef long long ll;

vector<ll> ans2(ll N,ll Q,vector<pair<ll,ll>> A,vector<ll> X){
    vector<ll> x(N+1,0),w(N+1,0),Sxw(N+2,0),Sw(N+2,0);
    sort(All(A));
    REP(i,N){
        x[i+1]=A[i].first;
        w[i+1]=A[i].second;
    }
    REP(i,N+1){
        Sxw[i+1]=Sxw[i]+x[i]*w[i];
        Sw[i+1]=Sw[i]+w[i];
    }
    vector<ll> res;
    REP(q,Q){
        int k=lower_bound(All(x),X[q])-x.begin();
        res.push_back(Sxw[N+1]-2*Sxw[k]+X[q]*(2*Sw[k]-Sw[N+1]));
    }
    return res;
}

int main(){
    ll N,Q;cin>>N>>Q;
    assert(1<=N&&N<=100000);
    assert(1<=Q&&Q<=100000);
    vector<pair<ll,ll>> A(N);
    set<int> appeared;
    REP(i,N){
        cin>>A[i].first>>A[i].second;
        assert(1<=A[i].first&&A[i].first<=1000000000);
        assert(1<=A[i].second&&A[i].second<=10000);
        assert(!appeared.count(A[i].first));
        appeared.insert(A[i].first);
    }
    appeared.clear();
    vector<ll> X(Q);
    REP(i,Q){
        cin>>X[i];
        assert(1<=X[i]&&X[i]<=1000000000);
        assert(!appeared.count(X[i]));
        appeared.insert(X[i]);
    }

    vector<ll> ans=ans2(N,Q,A,X);
    REP(i,Q){
        cout<<ans[i]<<endl;
    }
}
0