Submission #400509

#	Time	Username	Problem	Language	Result	Execution time	Memory
400509		Hazem	Fireworks (APIO16_fireworks)	C++14	19 / 100	36 ms	5844 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];
set<LL>dis;

void dfs(int i,LL d){

    if(adj[i].empty()){
        dis.insert(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 = LINF,val = 0;
    for(int i=0;i<=300;i++){
        if(dp[1][i]<ans)ans = dp[1][i],val = i;
    }

    if(dis.find(val)==dis.end())
        return 0;
        
    printf("%lld\n",ans);
}   

Compilation message (stderr)

fireworks.cpp: In function 'int main()':
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: 0 / 7
• Subtask 2: 19 / 19
• Subtask 3: 0 / 29
• Subtask 4: 0 / 45