CSES - Datatähti 2023 alku - Results
Submission details
Task:Sadonkorjuu
Sender:Luc
Submission time:2022-11-13 22:36:35 +0200
Language:C++11
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.00 s1, 2details
#20.00 s1, 2details
#30.00 s1, 2details
#40.00 s1, 2details
#50.00 s1, 2details
#60.00 s1, 2details
#70.00 s2details
#80.00 s1, 2details
#90.00 s2details
#100.00 s1, 2details
#110.00 s2details
#120.00 s2details
#130.00 s2details
#140.00 s2details
#150.00 s1, 2details
#160.00 s1, 2details
#170.00 s1, 2details
#180.00 s1, 2details
#190.00 s1, 2details
#200.00 s1, 2details
#210.00 s2details
#220.00 s2details
#230.00 s2details
#240.00 s1, 2details
#250.00 s2details
#260.00 s1, 2details
#270.00 s2details
#280.00 s1, 2details
#290.00 s2details
#300.00 s1, 2details
#310.00 s2details

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:41:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   41 |         freopen("lavbb", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~
input/code.cpp:44:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   44 |         scanf("%d", &n);
      |         ~~~~~^~~~~~~~~~
input/code.cpp:51:22: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   51 |                 scanf("%d", &x);
      |                 ~~~~~^~~~~~~~~~
input/code.cpp:60:22: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   60 |                 scanf("%d %d %d", &x, &y, &z);
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~~~~...

Code

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

stack<pair<int, int>> a;

ll processC(vector<ll> * sp, vector<bool> * field, vector<vector<pair<int, int>>> * conn, vector<bool> * pross, int c, int pc, ll cp){
	if(!(field[0][c])){ 
		if(!pross[0][c]){ 
			for(vector<pair<int,int>>::iterator it = conn[0][c].begin(); it!= conn[0][c].end(); it++){
				if(it->first==pc) continue;
				if(!field[0][it->first]) a.push({it->first,c}); 
				else processC(sp,field,conn,pross,it->first, c, 0+(ll)it->second);
			}
			pross[0][c]=1; 
		}
		return 0ll;
	}
	int t; 
	bool b = 0; 
	if( (!pross[0][c]) || sp[0][c]>cp){ 
		for(vector<pair<int,int>>::iterator it = conn[0][c].begin(); it!= conn[0][c].end(); it++){
			if(it->first==pc) continue;
			t=processC(sp,field,conn,pross,it->first, c, cp+(ll)it->second)+(ll)it->second;
			if(t<cp){ 
				cp=t;
				b = 1; 
			}
		}
		if(b) processC(sp,field,conn,pross,c, pc, cp); 
		
		pross[0][c]=1;
		sp[0][c]=cp;
	}
	return sp[0][c];
}

int main(){
	freopen("lavbb", "r", stdin);
	int n, i = 0, x, y, z;
	
	scanf("%d", &n);

	vector<ll> sp(n+1, 2999999999ll);
	
	vector<bool> field(n+1, 0); 

	while(i++<n){
		scanf("%d", &x);
		field[i]=(bool)x;
	}
	
	vector<vector<pair<int, int>>> conn(n+1);
	vector<bool> pross(n+1, 0);

	i=1;
	while(i++<n){
		scanf("%d %d %d", &x, &y, &z);
		conn[x].push_back({y,z});
		conn[y].push_back({x,z});
	}

	a.push({1, 0});
	pair<int, int> t; 

	while(!a.empty()){
		t=a.top();
		a.pop();
		processC(&sp,&field,&conn,&pross,t.first,t.second,2999999999ll);
	}
	
	i=0;
	ll r = 0; 

	while(i++<n) if(field[i])r+=sp[i];
	printf("%lld", r);
}

Test details

Test 1

Group: 1, 2

Verdict:

input
1
0

correct output
0

user output
(empty)

Test 2

Group: 1, 2

Verdict:

input
5
0 0 0 0 0
1 2 1
2 3 2
3 4 3
...

correct output
0

user output
(empty)

Test 3

Group: 1, 2

Verdict:

input
4
1 0 1 1
1 2 10
2 3 20
2 4 30

correct output
60

user output
(empty)

Test 4

Group: 1, 2

Verdict:

input
5
0 1 1 1 0
1 2 10
2 3 20
3 4 30
...

correct output
80

user output
(empty)

Test 5

Group: 1, 2

Verdict:

input
5
0 1 0 1 1
1 2 1
2 3 5
3 4 3
...

correct output
6

user output
(empty)

Test 6

Group: 1, 2

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5506363

user output
(empty)

Test 7

Group: 2

Verdict:

input
200000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
1795118520

user output
(empty)

Test 8

Group: 1, 2

Verdict:

input
1000
0 0 1 0 1 1 0 1 0 1 1 0 0 0 1 ...

correct output
293576

user output
(empty)

Test 9

Group: 2

Verdict:

input
200000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
816932444

user output
(empty)

Test 10

Group: 1, 2

Verdict:

input
1000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

correct output
3089

user output
(empty)

Test 11

Group: 2

Verdict:

input
200000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

correct output
40839

user output
(empty)

Test 12

Group: 2

Verdict:

input
200000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5683983203973

user output
(empty)

Test 13

Group: 2

Verdict:

input
200000
0 1 1 1 1 1 1 0 0 0 1 1 0 1 0 ...

correct output
58572993

user output
(empty)

Test 14

Group: 2

Verdict:

input
200000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

correct output
32755

user output
(empty)

Test 15

Group: 1, 2

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
126238345

user output
(empty)

Test 16

Group: 1, 2

Verdict:

input
1000
0 0 0 1 0 1 1 1 0 0 1 0 1 1 0 ...

correct output
278678

user output
(empty)

Test 17

Group: 1, 2

Verdict:

input
1000
1 0 0 0 1 0 0 0 1 0 0 0 0 0 0 ...

correct output
34929

user output
(empty)

Test 18

Group: 1, 2

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
1543963

user output
(empty)

Test 19

Group: 1, 2

Verdict:

input
1000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

correct output
39606

user output
(empty)

Test 20

Group: 1, 2

Verdict:

input
1000
1 0 1 0 1 0 0 0 0 1 1 0 0 0 1 ...

correct output
321598

user output
(empty)

Test 21

Group: 2

Verdict:

input
200000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
978670626

user output
(empty)

Test 22

Group: 2

Verdict:

input
200000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

correct output
375218

user output
(empty)

Test 23

Group: 2

Verdict:

input
200000
1 1 1 1 0 0 0 0 0 1 0 1 0 1 1 ...

correct output
60422556

user output
(empty)

Test 24

Group: 1, 2

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
291990

user output
(empty)

Test 25

Group: 2

Verdict:

input
200000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
59607954

user output
(empty)

Test 26

Group: 1, 2

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
990

user output
(empty)

Test 27

Group: 2

Verdict:

input
200000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
199982

user output
(empty)

Test 28

Group: 1, 2

Verdict:

input
1000
0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
7987

user output
(empty)

Test 29

Group: 2

Verdict:

input
200000
0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
3137875

user output
(empty)

Test 30

Group: 1, 2

Verdict:

input
1000
0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
4657693

user output
(empty)

Test 31

Group: 2

Verdict:

input
200000
0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
1652889357

user output
(empty)