제출 #939755

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
939755		vjudge1	Chase (CEOI17_chase)	C++17	70 / 100	202 ms	98496 KiB

chase

#include <bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
//using namespace __gnu_pbds;
 
bool YES(bool f){ if(f) cout << "Yes\n" ; else cout << "No\n" ; return f ; }
void YES(){YES(1);}
void NO(){YES(0);}
void fopn(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
//#define ordered_set tree<int, null_type,less_equal<int>, rb_tree_tag,tree_order_statistics_node_update>
#define ios ios_base::sync_with_stdio(0) ; cin.tie(0) ; cout.tie(0);
#define int long long
#define ld long double
#define pii pair <int , int>
#define all(x) x.begin() , x.end()
#define ff first
#define ss second
#define endl '\n'
 
const int N = 2e5 + 5 ;
const int inf = 1e16 ;
const int mod = 1e9 + 8 ;
const double eps = 1e-8 ;
 
template <class T>
bool chmax( T& x , const T& y ){
  bool f = 0 ;
  if ( x < y ) x = y , f = 1 ;
  return f ;
}
template <class T>
bool chmin( T &x , const T &y ){
  bool f = 0 ;
  if ( x > y ) x = y , f = 1 ;
  return f ;
}
 
//code
 
int n , m , ans ;
int a[N] , dp[N][103] , c[N] ;
vector <int> g[N] ;
bool vis[N] ;

void dfs ( int v , int p ){
	vis[v] = 1 ;
	for ( int i = 1 ; i <= m ; i ++ ){
		if ( dp[p][i] == -1 ) break ;
		chmax ( dp[v][i] , max(dp[p][i-1]+c[v]-a[p] , dp[p][i]) ) ;
		chmax ( ans , dp[v][i] ) ;
	}
	for ( auto to : g[v] ){
		if ( !vis[to] ){
			dfs(to,v) ;
		}
		
	}
}

void solve(){
	
	cin >> n >> m ;
	for ( int i = 1 ; i <= n ; i ++ ) cin >> a[i] ;
	for ( int i = 1 ; i <= n ; i ++ ) for ( int j = 0 ; j <= m ; j ++ ) dp[i][j] = -1 ;
	for ( int i = 1 ; i < n ; i ++ ){
		int x , y ; cin >> x >> y ;
		g[x].push_back(y) ;
		g[y].push_back(x) ;
		c[x] += a[y] ;
		c[y] += a[x] ;
	}
	if ( n <= 1000 ){
		for ( int i = 1 ; i <= n ; i ++ ){
			dfs(i,0) ;
			for ( int j = 1 ; j <= n ; j ++ ) vis[j] = 0 ;
			for ( int j = 1 ; j <= n ; j ++ ) for ( int l = 0 ; l <= m ; l ++ ) dp[j][l] = -1 ;
		}
	}
	else dfs(1,0) ;
	cout << ans ;
}

signed main(){
    ios ;
	int t = 1 ;
	//cin >> t ;
	while ( t -- ) solve() ;
}

컴파일 시 표준 에러 (stderr) 메시지

chase.cpp: In function 'void fopn(std::string)':
chase.cpp:10:31: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   10 | void fopn(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                        ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
chase.cpp:10:72: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   10 | void fopn(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                                                                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 20 / 20
• Subtask 2: 20 / 20
• Subtask 3: 30 / 30
• Subtask 4: 0 / 30