HackerRank Weekly Challenges - Week 2 2問目 - Cut the tree

Source

HackerRank Weekly Challenges - Week 2 2問目
problem statement(コンテストページ)
problem statement

問題概要

$N$ ノードからなる無向木が与えられる.
各ノード $k$ には重さ $W_k$ が付けられている.
$1$ 本枝を削ると木は $2$ つの部分にわかれるが,各部分のノードの重さの和ができるだけ同じにしたい.
各部分のノードの重さの和の差の絶対値の最小値を求める問題.

解法

適当なノードを根にして根付き木で考える.
葉に近いノードから順番に,各ノードについて,そのノードと,その子孫の重さの和を計算する.これはDFSなどで計算できる.
すると,各ノードの,その親をつなぐノードを切った時の各部分のノードの重さの和を簡単に計算することができる.
全体で時間計算量 $O(N)$.

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void reader(int *x, int *y){reader(x);reader(y);}
void writer(int x, char c){int i,sz=0,m=0;char buf[10];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}

int N;
vector<int> edge[100001];
int w[100001], sum[100001];

void dfs(int node, int up){
  int i, k;
  sum[node] = w[node];
  rep(k,edge[node].size()){
    i = edge[node][k];
    if(i==up) continue;
    dfs(i, node);
    sum[node] += sum[i];
  }
}

int main(){
  int i, j, k;
  int res, tmp;

  reader(&N);
  rep(i,N) reader(w+i);
  REP(k,1,N){
    reader(&i, &j);
    i--; j--;
    edge[i].push_back(j);
    edge[j].push_back(i);
  }

  dfs(0, -1);
  res = sum[0];
  REP(i,1,N) res = min(res, abs(sum[0]-2*sum[i]));

  writer(res, '\n');

  return 0;
}

Current time: 2017年09月21日05時04分39秒
Last modified: 2014年09月22日23時36分33秒 (by laycrs)
Tags: Competitive_Programming HackerRank HackerRank_w2
トップページに戻る

Logged in as: unknown user (not login)

ログイン: