oj.uz

Submission #598519

# Submission time Handle Problem Language Result Execution time Memory
598519 2022-07-18T12:53:57 Z AGE Transport (COCI19_transport) C++14
65 / 130
 729 ms 38200 KB 
#include<bits/stdc++.h>
#define F first
#define S second
#define int long long
#define pb push_back

using namespace std;
const int N=2e5+10,M=2e3,mod=1e9+7;
int cost[N],seg[N*4],ans[N];
int n;
vector<pair<int,int>>adj[N];
map<pair<int,int>,int>mp;

void dfs(int node,int par){

    for(auto x:adj[node]){

        if(x.F==par)
            continue;

        ans[x.F]=ans[node]+cost[node]-mp[{node,x.F}];
        if(ans[x.F]<0)
            continue;

        dfs(x.F,node);

    }

}

void build(int v,int tl,int tr){

    if(tl==tr){
        seg[v]=ans[tl];
        return ;
    }

    int tm=(tl+tr)/2;

    build(v*2,tl,tm);
    build(v*2+1,tm+1,tr);

    seg[v]=min(seg[v*2],seg[v*2+1]);

}

int get(int v,int tl,int tr,int l,int r){

    if(tl>r||tr<l){
        return 1e18;
    }

    if(tl>=l&&tr<=r)
        return seg[v];

    int tm=(tl+tr)/2;

    return min(get(v*2,tl,tm,l,r),get(v*2+1,tm+1,tr,l,r));

}

bool ok(int mid,int val,int start){

    if(start+mid-1>n)
        return 0;

    int ans1=get(1,1,n,start,start+mid-1);
    return ans1>=val;

}
main()
{
    cin>>n;
        for(int i=1;i<=n;i++)
            cin>>cost[i];

        for(int i=0;i<n-1;i++){
            int x,y,z;
            cin>>x>>y>>z;

            adj[x].pb({y,z});
            adj[y].pb({x,z});
            mp[{x,y}]=z;
            mp[{y,x}]=z;

        }

    if(n<=5000){
        int anss=0;

        for(int i=1;i<=n;i++){

            for(int j=1;j<=n;j++)
                ans[j]=-1e18;

            ans[i]=0;
            dfs(i,0);


            for(int j=1;j<=n;j++)
                if(ans[j]!=1e18&&ans[j]>=0&&i!=j)
                    anss++;

        }

        cout<<anss<<endl;
    }


    else{
       ans[1]=0;

    for(int i=1;i<=n;i++){
        ans[i+1]=ans[i]+cost[i]-mp[{i,i+1}];
    }

    build(1,1,n);
    int anss=0;

    for(int i=1;i<=n;i++){

        int val=ans[i];
        int l=0,r=n+1;

        while(l<r){

            int mid=(l+r+1)/2;
            if(ok(mid,val,i))
                l=mid;
            else
                r=mid-1;

        }

        anss+=l-1;

    }

    reverse(cost+1,cost+n+1);

    ans[1]=0;

    for(int i=1;i<=n;i++){
        ans[i+1]=ans[i]+cost[i]-mp[{n-i+1,n-i}];
    }

    build(1,1,n);

    for(int i=1;i<=n;i++){

        int val=ans[i];
        int l=1,r=n+1;

        while(l<r){

            int mid=(l+r+1)/2;
            if(ok(mid,val,i))
                l=mid;
            else
                r=mid-1;

        }

        anss+=l-1;

    }

    cout<<anss<<endl;
    }
    return 0;
}

Compilation message

transport.cpp:71:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   71 | main()
      | ^~~~

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 210 ms 5588 KB Output is correct
2 Correct 133 ms 5784 KB Output is correct

Subtask #2
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 507 ms 6300 KB Output is correct
2 Correct 39 ms 5964 KB Output is correct

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 254 ms 14184 KB Output is correct
2 Correct 249 ms 13340 KB Output is correct

Subtask #4
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 422 ms 17224 KB Output is correct
2 Correct 464 ms 19532 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 541 ms 22332 KB Output is correct
2 Correct 729 ms 25732 KB Output is correct

Subtask #6
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 150 ms 14100 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 271 ms 20336 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 385 ms 25284 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #9
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 513 ms 31948 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 673 ms 38200 KB Output isn't correct
2 Halted 0 ms 0 KB -