Submission #400515

#	Time	Username	Problem	Language	Result	Execution time	Memory
400515		Hazem	Fireworks (APIO16_fireworks)	C++14	7 / 100	12 ms	5280 KiB

fireworks

#include <bits/stdc++.h>
using namespace std;
 
#define LL long long
#define F first
#define S second
#define pii pair<int,int>
#define piii pair<pair<int,int>,int>

const int N = 2e5+10;
const int M = 2e3+10;
const LL INF = 1e9;
const LL LINF = 1e14;
const LL MOD = 1e9+7;
const double PI = 3.141592653589793;

vector<pair<int,LL>>adj[N];
LL dp[N][350];
vector<LL>vec;

void dfs(int i,LL d){

    if(adj[i].empty()){
        vec.push_back(d);
        for(int j=1;j<=300;j++)
            dp[i][j] = LINF;
    }

    for(auto x:adj[i]){
        
        dfs(x.F,d+x.S);

        for(int j=0;j<=300;j++){
     
            LL mn = LINF;
            for(int k=0;k<=j;k++)
                mn = min(mn,dp[x.F][k]+abs(x.S-(abs(k-j))));

            dp[i][j] += mn;
        }
    }
}

int main(){

    //freopen("out.txt","w",stdout);

    int n,m;
    scanf("%d%d",&n,&m);

    for(int i=2;i<=n+m;i++){
        LL v,w;
        scanf("%lld%lld",&v,&w);
        adj[v].push_back({i,w});
    }

    dfs(1,0);
    LL ans = 0,val = 0;
    sort(vec.begin(),vec.end());
    for(auto x:vec)
        ans += abs(x-vec[vec.size()/2]);

    printf("%lld\n",ans);
}   

Compilation message (stderr)

fireworks.cpp: In function 'int main()':
fireworks.cpp:58:16: warning: unused variable 'val' [-Wunused-variable]
   58 |     LL ans = 0,val = 0;
      |                ^~~
fireworks.cpp:49:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   49 |     scanf("%d%d",&n,&m);
      |     ~~~~~^~~~~~~~~~~~~~
fireworks.cpp:53:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   53 |         scanf("%lld%lld",&v,&w);
      |         ~~~~~^~~~~~~~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 0 / 19
• Subtask 3: 0 / 29
• Subtask 4: 0 / 45