oj.uz

Submission #80158

# Submission time Handle Problem Language Result Execution time Memory
80158 2018-10-19T08:26:29 Z farukkastamonuda Dostavljač (COCI18_dostavljac) C++14
140 / 140
 218 ms 2988 KB 
#include <bits/stdc++.h>
#define fi first
#define se second
#define lo long long 
#define inf 1000000000
#define md 1000000007
#define li 505
#define mp make_pair
#define pb push_back
using namespace std;
int n,T,x,y,dp[li][li],A[li],en[li][li];
vector<int> v[li];
void dfs(int node,int ata){
	for(int i=0;i<(int)v[node].size();i++){
		int go=v[node][i];
		if(go==ata) continue;
		dfs(go,node);
	}
	for(int i=0;i<(int)v[node].size();i++){
		int go=v[node][i];
		if(go==ata) continue;
		for(int k=T;k>=3;k--){
			for(int j=1;j+2<=k;j++){
				en[node][k]=max(en[node][k],en[node][k-j-2]+dp[go][j]);
			}
		}
		for(int k=T;k>=2;k--){
			for(int j=1;j+1<=k;j++){
				en[node][k]=max(en[node][k],dp[node][k-j-1]+en[go][j]);
			}
		}
		for(int k=T;k>=3;k--){
			for(int j=1;j+2<=k;j++){
				dp[node][k]=max(dp[node][k],dp[node][k-j-2]+dp[go][j]);
			}
		}
	}
	for(int j=T;j>=1;j--){
		dp[node][j]=max(dp[node][j],dp[node][j-1]+A[node]);
		en[node][j]=max(en[node][j],en[node][j-1]+A[node]);
	}
}
int main(){
	scanf("%d %d",&n,&T);
	for(int i=1;i<=n;i++) scanf("%d",&A[i]);
	for(int i=1;i<n;i++){
		scanf("%d %d",&x,&y);
		v[x].pb(y);
		v[y].pb(x);
	}
	dfs(1,0);
	printf("%d\n",en[1][T]);
	return 0;
}

Compilation message

dostavljac.cpp: In function 'int main()':
dostavljac.cpp:44:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d %d",&n,&T);
  ~~~~~^~~~~~~~~~~~~~~
dostavljac.cpp:45:29: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  for(int i=1;i<=n;i++) scanf("%d",&A[i]);
                        ~~~~~^~~~~~~~~~~~
dostavljac.cpp:47:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d %d",&x,&y);
   ~~~~~^~~~~~~~~~~~~~~

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 508 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 572 KB Output is correct

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 780 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 1020 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 1064 KB Output is correct
2 Correct 5 ms 1068 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 1328 KB Output is correct
2 Correct 36 ms 1528 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 82 ms 1576 KB Output is correct
2 Correct 21 ms 1580 KB Output is correct

Subtask #8
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 1968 KB Output is correct
2 Correct 167 ms 2080 KB Output is correct

Subtask #9
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 134 ms 2520 KB Output is correct
2 Correct 51 ms 2676 KB Output is correct

Subtask #10
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 218 ms 2988 KB Output is correct
2 Correct 26 ms 2988 KB Output is correct