답안 #565192

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
565192	2022-05-20T12:30:28 Z	CSQ31	The Xana coup (BOI21_xanadu)	C++17	100 / 100	95 ms	23168 KB

xanadu

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5+1;
vector<int>adj[MAXN];
typedef long long int ll;
#define owo ios_base::sync_with_stdio(0);cin.tie(0);
ll dp[2][2][MAXN];
int a[MAXN];
//toggled , on/off
void print(int v){
	cout<<v<<'\n';
	cout<<"00 "<<dp[0][0][v]<<'\n';
	cout<<"01 "<<dp[0][1][v]<<'\n';
	cout<<"10 "<<dp[1][0][v]<<'\n';
	cout<<"11 "<<dp[1][1][v]<<'\n';
}
void dfs(int v,int u){
	if((int)(adj[v].size()) == 1 && v>1){
		dp[0][a[v]][v] = 0;
		dp[1][a[v]^1][v] = 1;
		//print(v);
		return;
	}
	vector<ll>d;
	ll tot = 0; //all children must be off
	for(int x:adj[v]){
		if(x==u)continue;
		dfs(x,v);	
		tot+=dp[0][0][x];
		d.push_back(dp[1][0][x] - dp[0][0][x]); 
	}
	sort(d.begin(),d.end());
	ll sum = 0;
	int m = d.size();
	for(int i=0;i<=m;i++){
		if(i)sum+=d[i-1];
		if((i%2) == a[v])dp[0][0][v] = min(dp[0][0][v],tot+sum); //toggle parity a[i] times
		else dp[0][1][v] = min(dp[0][1][v],tot+sum);
	}
	/////
	tot = 0;
	d.clear();
	//all children has to be on
	for(int x:adj[v]){
		if(x==u)continue;
		tot+=dp[0][1][x];
		d.push_back(dp[1][1][x] - dp[0][1][x]); 
	}
	sort(d.begin(),d.end());
	sum = 0;
	m = d.size();
	for(int i=0;i<=m;i++){
		if(i)sum+=d[i-1];
		if((i%2) != a[v])dp[1][0][v] = min(dp[1][0][v],tot+sum+1);
		else dp[1][1][v] = min(dp[1][1][v],tot+sum+1);
	}
	for(int i=0;i<4;i++)dp[i&1][i>>1][v] = min(dp[i&1][i>>1][v],(ll)(1e9));
	//print(v);
}
int main()
{
	owo
	int n;
	cin>>n;
	for(int i=1;i<n;i++){
		int v,u;
		cin>>v>>u;
		adj[v].push_back(u);
		adj[u].push_back(v);
	}
	for(int i=1;i<=n;i++){
		cin>>a[i];
		dp[0][0][i] = dp[0][1][i] = dp[1][0][i] = dp[1][1][i] = 1e9;
	}
	dfs(1,0);
	ll ans = min(dp[1][0][1],dp[0][0][1]);
	if(ans==1e9)cout<<"impossible"<<'\n';
	else cout<<ans<<'\n';
}

Subtask #1
5.0 / 5.0

#	결과	실행 시간	메모리	Grader output
1	Correct	2 ms	2680 KB	Output is correct
2	Correct	3 ms	2644 KB	Output is correct
3	Correct	3 ms	2644 KB	Output is correct
4	Correct	2 ms	2644 KB	Output is correct
5	Correct	3 ms	2644 KB	Output is correct

Subtask #2
15.0 / 15.0

#	결과	실행 시간	메모리	Grader output
1	Correct	2 ms	2680 KB	Output is correct
2	Correct	3 ms	2644 KB	Output is correct
3	Correct	3 ms	2644 KB	Output is correct
4	Correct	2 ms	2644 KB	Output is correct
5	Correct	3 ms	2644 KB	Output is correct
6	Correct	2 ms	2644 KB	Output is correct
7	Correct	2 ms	2644 KB	Output is correct
8	Correct	2 ms	2644 KB	Output is correct
9	Correct	2 ms	2608 KB	Output is correct
10	Correct	2 ms	2644 KB	Output is correct
11	Correct	2 ms	2644 KB	Output is correct
12	Correct	2 ms	2644 KB	Output is correct
13	Correct	2 ms	2676 KB	Output is correct
14	Correct	2 ms	2644 KB	Output is correct
15	Correct	2 ms	2644 KB	Output is correct
16	Correct	2 ms	2644 KB	Output is correct

Subtask #3
10.0 / 10.0

#	결과	실행 시간	메모리	Grader output
1	Correct	83 ms	23132 KB	Output is correct
2	Correct	72 ms	22728 KB	Output is correct
3	Correct	66 ms	23128 KB	Output is correct

Subtask #4
40.0 / 40.0

#	결과	실행 시간	메모리	Grader output
1	Correct	65 ms	22976 KB	Output is correct
2	Correct	66 ms	22840 KB	Output is correct
3	Correct	84 ms	23052 KB	Output is correct
4	Correct	63 ms	9804 KB	Output is correct
5	Correct	63 ms	10484 KB	Output is correct
6	Correct	92 ms	10664 KB	Output is correct
7	Correct	3 ms	2644 KB	Output is correct
8	Correct	30 ms	5392 KB	Output is correct
9	Correct	84 ms	10644 KB	Output is correct
10	Correct	95 ms	10796 KB	Output is correct
11	Correct	81 ms	11612 KB	Output is correct
12	Correct	94 ms	11908 KB	Output is correct
13	Correct	68 ms	10096 KB	Output is correct
14	Correct	70 ms	10652 KB	Output is correct

Subtask #5
30.0 / 30.0

#	결과	실행 시간	메모리	Grader output
1	Correct	2 ms	2680 KB	Output is correct
2	Correct	3 ms	2644 KB	Output is correct
3	Correct	3 ms	2644 KB	Output is correct
4	Correct	2 ms	2644 KB	Output is correct
5	Correct	3 ms	2644 KB	Output is correct
6	Correct	2 ms	2644 KB	Output is correct
7	Correct	2 ms	2644 KB	Output is correct
8	Correct	2 ms	2644 KB	Output is correct
9	Correct	2 ms	2608 KB	Output is correct
10	Correct	2 ms	2644 KB	Output is correct
11	Correct	2 ms	2644 KB	Output is correct
12	Correct	2 ms	2644 KB	Output is correct
13	Correct	2 ms	2676 KB	Output is correct
14	Correct	2 ms	2644 KB	Output is correct
15	Correct	2 ms	2644 KB	Output is correct
16	Correct	2 ms	2644 KB	Output is correct
17	Correct	83 ms	23132 KB	Output is correct
18	Correct	72 ms	22728 KB	Output is correct
19	Correct	66 ms	23128 KB	Output is correct
20	Correct	65 ms	22976 KB	Output is correct
21	Correct	66 ms	22840 KB	Output is correct
22	Correct	84 ms	23052 KB	Output is correct
23	Correct	63 ms	9804 KB	Output is correct
24	Correct	63 ms	10484 KB	Output is correct
25	Correct	92 ms	10664 KB	Output is correct
26	Correct	3 ms	2644 KB	Output is correct
27	Correct	30 ms	5392 KB	Output is correct
28	Correct	84 ms	10644 KB	Output is correct
29	Correct	95 ms	10796 KB	Output is correct
30	Correct	81 ms	11612 KB	Output is correct
31	Correct	94 ms	11908 KB	Output is correct
32	Correct	68 ms	10096 KB	Output is correct
33	Correct	70 ms	10652 KB	Output is correct
34	Correct	2 ms	2644 KB	Output is correct
35	Correct	2 ms	2684 KB	Output is correct
36	Correct	2 ms	2644 KB	Output is correct
37	Correct	2 ms	2604 KB	Output is correct
38	Correct	2 ms	2644 KB	Output is correct
39	Correct	2 ms	2644 KB	Output is correct
40	Correct	2 ms	2644 KB	Output is correct
41	Correct	2 ms	2644 KB	Output is correct
42	Correct	2 ms	2644 KB	Output is correct
43	Correct	2 ms	2644 KB	Output is correct
44	Correct	2 ms	2692 KB	Output is correct
45	Correct	59 ms	23048 KB	Output is correct
46	Correct	70 ms	22844 KB	Output is correct
47	Correct	58 ms	23168 KB	Output is correct
48	Correct	58 ms	9820 KB	Output is correct
49	Correct	65 ms	10612 KB	Output is correct
50	Correct	95 ms	10656 KB	Output is correct
51	Correct	2 ms	2684 KB	Output is correct
52	Correct	33 ms	5312 KB	Output is correct
53	Correct	75 ms	10668 KB	Output is correct
54	Correct	78 ms	10792 KB	Output is correct
55	Correct	80 ms	11640 KB	Output is correct
56	Correct	83 ms	11940 KB	Output is correct
57	Correct	63 ms	10060 KB	Output is correct
58	Correct	80 ms	10756 KB	Output is correct
59	Correct	20 ms	5332 KB	Output is correct
60	Correct	66 ms	9868 KB	Output is correct
61	Correct	61 ms	10664 KB	Output is correct
62	Correct	63 ms	10732 KB	Output is correct
63	Correct	60 ms	10708 KB	Output is correct
64	Correct	61 ms	10680 KB	Output is correct
65	Correct	45 ms	11152 KB	Output is correct
66	Correct	54 ms	11096 KB	Output is correct
67	Correct	53 ms	12144 KB	Output is correct
68	Correct	40 ms	12220 KB	Output is correct
69	Correct	42 ms	12232 KB	Output is correct
70	Correct	42 ms	12116 KB	Output is correct

답안 #565192

Compilation message

Subtask #1 5.0 / 5.0

Subtask #2 15.0 / 15.0

Subtask #3 10.0 / 10.0

Subtask #4 40.0 / 40.0

Subtask #5 30.0 / 30.0

Subtask #1
5.0 / 5.0

Subtask #2
15.0 / 15.0

Subtask #3
10.0 / 10.0

Subtask #4
40.0 / 40.0

Subtask #5
30.0 / 30.0