제출 #1137464

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1137464		icoder178	Harbingers (CEOI09_harbingers)	C++20	40 / 100	80 ms	37192 KiB

harbingers

#include<iostream>
#include<algorithm>
using namespace std;
#define int long long
const int maxn = 100005;
int dp[maxn];
signed n,dst[maxn],rmv[maxn],s[maxn],v[maxn],connl[maxn],connr[maxn],ncnt = 0,chcnt = 0;
pair<signed,int> cht[maxn];
pair<signed,signed> nconn[maxn];
pair<signed,pair<signed,signed> > conn[maxn*2];
inline void fdp(int pos) {
    int l = -5, r = chcnt+5;
    while(l+1 < r) {
        int mid = (l+r)/2;
        bool isDecr;
        if(mid < 0) isDecr = true;
        else if(mid >= chcnt-1) isDecr = false;
        else {
            int vl1 = cht[mid].second-(int) v[pos]*(int) cht[mid].first;
            int vl2 = cht[mid+1].second-(int) v[pos]*(int) cht[mid+1].first;
            isDecr = (vl2 < vl1);
        }
        if(isDecr == true) l = mid;
        else r = mid;
    }
    dp[pos] = cht[r].second-(int) v[pos]*(int) cht[r].first+(int) v[pos]*(int) dst[pos]+(int) s[pos];
}
inline void prmv(int pos) {
    int l = -5, r = chcnt+5;
    while(l+1 < r) {
        int mid = (l+r)/2;
        bool isV;
        if(mid < 0) isV = true;
        else if(mid >= (chcnt-1)) isV = false;
        else isV = (((__int128_t) dp[pos]-(__int128_t) cht[mid+1].second)
        *((__int128_t) cht[mid+1].first-(__int128_t) cht[mid].first) 
        > ((__int128_t) cht[mid+1].second-(__int128_t) cht[mid].second)
        *((__int128_t) dst[pos]-(__int128_t) cht[mid+1].first));
        if(isV) l = mid;
        else r = mid;
    }
    rmv[pos] = r+1;
}
void dfs(int pos, int prevP) {
    int ncl = ncnt;
    for(int i = connl[pos]; i < connr[pos]; i++) {
        int targP = conn[i].second.first;
        int newV = conn[i].second.second;
        if(targP != prevP) {
            dst[targP] = dst[pos]+newV;
            fdp(targP);
            prmv(targP);
            nconn[ncnt].first = rmv[targP];
            nconn[ncnt].second = targP;
            ncnt++;
        }
    }
    int ncr = ncnt;
    sort(nconn+ncl,nconn+ncr);
    for(int i = ncr-1; i >= ncl; i--) {
        while(chcnt > nconn[i].first) chcnt--;
        int targP = nconn[i].second;
        cht[chcnt].first = dst[targP];
        cht[chcnt].second = dp[targP];
        chcnt++;
        dfs(targP,pos);
        chcnt--;
    }
    ncnt = ncl;
}
signed main () {
    scanf("%d",&n);
    for(int i = 0; i < n-1; i++) {
        int a,b,c;
        scanf("%d%d%d",&conn[i*2].first,&conn[i*2].second.first,&conn[i*2].second.second);
        conn[i*2+1].first = conn[i*2].second.first;
        conn[i*2+1].second.first = conn[i*2].first;
        conn[i*2+1].second.second = conn[i*2].second.second;
    }
    sort(conn,conn+n*2-2);
    for(int i = 0; i < n*2-2; i++) {
        connr[conn[i].first] = i+1;
    }
    for(int i = n*2-3; i >= 0; i--) {
        connl[conn[i].first] = i;
    }
    for(int i = 2; i <= n; i++) {
        scanf("%d%d",&s[i],&v[i]);
    }
    chcnt++;
    dfs(1,-1);
    for(int i = 2; i <= n; i++) {
        printf("%lld ",dp[i]);
    }
    printf("\n");
    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

harbingers.cpp: In function 'int main()':
harbingers.cpp:72:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   72 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
harbingers.cpp:75:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   75 |         scanf("%d%d%d",&conn[i*2].first,&conn[i*2].second.first,&conn[i*2].second.second);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
harbingers.cpp:88:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   88 |         scanf("%d%d",&s[i],&v[i]);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~

• Subtask 1: 40 / 100