Submission #1214035

#TimeUsernameProblemLanguageResultExecution timeMemory
1214035trinm01Harbingers (CEOI09_harbingers)C++20
70 / 100
1096 ms22196 KiB
#pragma GCC target("avx")
#pragma GCC optimize(3)
#pragma GCC optimize("Ofast")
#pragma GCC optimize("inline")
#pragma GCC optimize("-fgcse")
#pragma GCC optimize("-fgcse-lm")
#pragma GCC optimize("-fipa-sra")
#pragma GCC optimize("-ftree-pre")
#pragma GCC optimize("-ftree-vrp")
#pragma GCC optimize("-fpeephole2")
#pragma GCC optimize("-ffast-math")
#pragma GCC optimize("-fsched-spec")
#pragma GCC optimize("unroll-loops")
#pragma GCC optimize("-falign-jumps")
#pragma GCC optimize("-falign-loops")
#pragma GCC optimize("-falign-labels")
#pragma GCC optimize("-fdevirtualize")
#pragma GCC optimize("-fcaller-saves")
#pragma GCC optimize("-fcrossjumping")
#pragma GCC optimize("-fthread-jumps")
#pragma GCC optimize("-funroll-loops")
#pragma GCC optimize("-fwhole-program")
#pragma GCC optimize("-freorder-blocks")
#pragma GCC optimize("-fschedule-insns")
#pragma GCC optimize("inline-functions")
#pragma GCC optimize("-ftree-tail-merge")
#pragma GCC optimize("-fschedule-insns2")
#pragma GCC optimize("-fstrict-aliasing")
#pragma GCC optimize("-fstrict-overflow")
#pragma GCC optimize("-falign-functions")
#pragma GCC optimize("-fcse-skip-blocks")
#pragma GCC optimize("-fcse-follow-jumps")
#pragma GCC optimize("-fsched-interblock")
#pragma GCC optimize("-fpartial-inlining")
#pragma GCC optimize("no-stack-protector")
#pragma GCC optimize("-freorder-functions")
#pragma GCC optimize("-findirect-inlining")
#pragma GCC optimize("-fhoist-adjacent-loads")
#pragma GCC optimize("-frerun-cse-after-loop")
#pragma GCC optimize("inline-small-functions")
#pragma GCC optimize("-finline-small-functions")
#pragma GCC optimize("-ftree-switch-conversion")
#pragma GCC optimize("-foptimize-sibling-calls")
#pragma GCC optimize("-fexpensive-optimizations")
#pragma GCC optimize("-funsafe-loop-optimizations")
#pragma GCC optimize("inline-functions-called-once")
#pragma GCC optimize("-fdelete-null-pointer-checks")
#include <bits/stdc++.h>
using namespace std;

#define int long long
#define ll long long
#define FOR(i, l, r) for (int i = (l); i <= (r); i++)
#define FOD(i, r, l) for (int i = (r); i >= (l); i--)
#define fi first
#define se second
#define pii pair<int, int>

const ll mod = 1e9 + 7;
const ll MAXN = 1e5 + 1;
const int oo = 1e18;
const int base = 320;

int n, s[MAXN], p[MAXN], d[MAXN];
vector<pii> adj[MAXN];
int xoa[MAXN];
int f[MAXN];
vector<int> A, B, xoaA, xoaB;
bool bad(int l1, int l2, int l3){
    return (B[l3]-B[l1])*(A[l1]-A[l2])<(B[l2]-B[l1])*(A[l1]-A[l3]);
}
void add(int a, int b, int u){
    // if(u==7){
    //     for(auto x:A){
    //         cout << x << ' ';
    //     }
    //     cout << '\n';
    // }
    A.push_back(a);
    B.push_back(b);
    
    while((int)A.size()>=3 && bad(A.size()-3, A.size()-2, A.size()-1)){
        xoaA.push_back(A[A.size()-2]);
        xoaB.push_back(B[B.size()-2]);
        xoa[u]++;
        A.erase(A.end()-2);
        B.erase(B.end()-2);
    }
    // if(u==9){
    //     for(auto x:A){
    //         cout << x << ' ';
    //     }
    //     cout << '\n';
    //     for(auto x:B){
    //         cout << x << ' ';
    //     }
    //     cout << '\n';
    // }
}
void hoi(int a, int b, int u){
    A.pop_back();
    B.pop_back();
    int cnt=0;
    while(cnt<xoa[u]){
        A.push_back(xoaA.back());
        B.push_back(xoaB.back());
        xoaA.pop_back();
        xoaB.pop_back();
        cnt++;
    }
    xoa[u]=0;
    // if(u==7){
    //     for(auto x:A){
    //         cout << x << ' ';
    //     }
    //     cout << '\n';
    // }
}
int query(int x){
    int l=1, r=(int)A.size()-1, id=1;
    if(r==-1){
        return 0;
    }
    if(r==0){
        return A[0]*x+B[0];
    }
    int ans=oo;
    while(l<=r){
        int mid=(l+r)/2;
        if(A[mid]*x+B[mid]<A[mid-1]*x+B[mid-1]){
            id=mid;
            l=mid+1;
            ans=min(ans, A[mid]*x+B[mid]);
        }
        else{
            r=mid-1;
            ans=min(ans, A[mid-1]+B[mid-1]);
        }
    }
    // if(x==6)cout << id << '\n';
    ans=min({ans, A[id]*x+B[id], A[id-1]*x+B[id-1]});
    if(id!=A.size()-1){
        ans=min(ans, A[id+1]*x+B[id+1]);
    }
    return ans;
}

void dfs2(int u, int last){
    if(u!=1){
        f[u]=query(p[u])+d[u]*p[u]+s[u];
        // if(u==4) cout << query(p[u]) << ' ';
    }
    add(-d[u], f[u], u);

    for(auto [v, c]:adj[u]){
        if(v==last) continue;
        d[v]=d[u]+c;
        dfs2(v, u);
    }

    hoi(-d[u], f[u], u);
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    if (fopen(".INP", "r"))
    {
        freopen(".INP", "r", stdin);
        freopen(".OUT", "w", stdout);
    }       

    cin >> n;
    FOR(i, 1, n-1){
        int u, v, c;
        cin >> u >> v >> c;
        adj[u].push_back({v, c});
        adj[v].push_back({u, c});
    }
    FOR(i, 2, n){
        cin >> s[i] >> p[i];    
    }

    dfs2(1, 1);
    FOR(i, 2, n){
        cout << f[i] << ' ';
    }

    return 0;
}

Compilation message (stderr)

harbingers.cpp:22:39: warning: bad option '-fwhole-program' to pragma 'optimize' [-Wpragmas]
   22 | #pragma GCC optimize("-fwhole-program")
      |                                       ^
harbingers.cpp:29:41: warning: bad option '-fstrict-overflow' to pragma 'optimize' [-Wpragmas]
   29 | #pragma GCC optimize("-fstrict-overflow")
      |                                         ^
harbingers.cpp:31:41: warning: bad option '-fcse-skip-blocks' to pragma 'optimize' [-Wpragmas]
   31 | #pragma GCC optimize("-fcse-skip-blocks")
      |                                         ^
harbingers.cpp:45:51: warning: bad option '-funsafe-loop-optimizations' to pragma 'optimize' [-Wpragmas]
   45 | #pragma GCC optimize("-funsafe-loop-optimizations")
      |                                                   ^
harbingers.cpp:69:32: warning: bad option '-fwhole-program' to attribute 'optimize' [-Wattributes]
   69 | bool bad(int l1, int l2, int l3){
      |                                ^
harbingers.cpp:69:32: warning: bad option '-fstrict-overflow' to attribute 'optimize' [-Wattributes]
harbingers.cpp:69:32: warning: bad option '-fcse-skip-blocks' to attribute 'optimize' [-Wattributes]
harbingers.cpp:69:32: warning: bad option '-funsafe-loop-optimizations' to attribute 'optimize' [-Wattributes]
harbingers.cpp:72:29: warning: bad option '-fwhole-program' to attribute 'optimize' [-Wattributes]
   72 | void add(int a, int b, int u){
      |                             ^
harbingers.cpp:72:29: warning: bad option '-fstrict-overflow' to attribute 'optimize' [-Wattributes]
harbingers.cpp:72:29: warning: bad option '-fcse-skip-blocks' to attribute 'optimize' [-Wattributes]
harbingers.cpp:72:29: warning: bad option '-funsafe-loop-optimizations' to attribute 'optimize' [-Wattributes]
harbingers.cpp:100:29: warning: bad option '-fwhole-program' to attribute 'optimize' [-Wattributes]
  100 | void hoi(int a, int b, int u){
      |                             ^
harbingers.cpp:100:29: warning: bad option '-fstrict-overflow' to attribute 'optimize' [-Wattributes]
harbingers.cpp:100:29: warning: bad option '-fcse-skip-blocks' to attribute 'optimize' [-Wattributes]
harbingers.cpp:100:29: warning: bad option '-funsafe-loop-optimizations' to attribute 'optimize' [-Wattributes]
harbingers.cpp:119:16: warning: bad option '-fwhole-program' to attribute 'optimize' [-Wattributes]
  119 | int query(int x){
      |                ^
harbingers.cpp:119:16: warning: bad option '-fstrict-overflow' to attribute 'optimize' [-Wattributes]
harbingers.cpp:119:16: warning: bad option '-fcse-skip-blocks' to attribute 'optimize' [-Wattributes]
harbingers.cpp:119:16: warning: bad option '-funsafe-loop-optimizations' to attribute 'optimize' [-Wattributes]
harbingers.cpp:148:26: warning: bad option '-fwhole-program' to attribute 'optimize' [-Wattributes]
  148 | void dfs2(int u, int last){
      |                          ^
harbingers.cpp:148:26: warning: bad option '-fstrict-overflow' to attribute 'optimize' [-Wattributes]
harbingers.cpp:148:26: warning: bad option '-fcse-skip-blocks' to attribute 'optimize' [-Wattributes]
harbingers.cpp:148:26: warning: bad option '-funsafe-loop-optimizations' to attribute 'optimize' [-Wattributes]
harbingers.cpp:164:13: warning: bad option '-fwhole-program' to attribute 'optimize' [-Wattributes]
  164 | signed main()
      |             ^
harbingers.cpp:164:13: warning: bad option '-fstrict-overflow' to attribute 'optimize' [-Wattributes]
harbingers.cpp:164:13: warning: bad option '-fcse-skip-blocks' to attribute 'optimize' [-Wattributes]
harbingers.cpp:164:13: warning: bad option '-funsafe-loop-optimizations' to attribute 'optimize' [-Wattributes]
harbingers.cpp: In function 'int main()':
harbingers.cpp:171:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  171 |         freopen(".INP", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~
harbingers.cpp:172:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  172 |         freopen(".OUT", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...