제출 #1273601

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1273601		mkkkkkkkk	트리 (IOI24_tree)	C++20	0 / 100	2097 ms	23820 KiB

tree

#include "tree.h"
#include <bits/stdc++.h>

using namespace std;

vector<int> adj[200001];
int arr[200001],arr2[200001];


void dfs(int i,long long& res,int l,int r)
{
    int sum=0;
    for(auto it : adj[i])
    {
        dfs(it,res,l,r);
        sum+=arr2[it];
    }
    if(adj[i].size()==0)
    {
        sum=l;
        res+=l*arr[i];
    }
    if(sum>r)
    {
        int br=sum-r;
        sum=r;
        res+=br*arr[i];
    }
    arr2[i]=l;
}

void init(std::vector<int> P, std::vector<int> W) {
  for(int i=1;i<P.size();i++)
  {
      adj[P[i]].push_back(i);
      arr[i]=W[i];
  }
  arr[0]=W[0];

}

long long query(int L, int R)
{
    long long res=0;
    dfs(0,res,L,R);
    return res;
}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 10
• Subtask 2: 0 / 13
• Subtask 3: 0 / 18
• Subtask 4: 0 / 7
• Subtask 5: 0 / 11
• Subtask 6: 0 / 22
• Subtask 7: 0 / 19