/ SeriousOJ /

Record Detail

System Error


  
# Status Time Cost Memory Cost
#1 System Error 1ms 764.0 KiB
#2 System Error 1ms 536.0 KiB
#3 System Error 1ms 532.0 KiB
#4 System Error 1ms 532.0 KiB
#5 System Error 21ms 2.652 MiB
#6 System Error 23ms 2.457 MiB
#7 System Error 22ms 2.77 MiB
#8 System Error 2ms 532.0 KiB
#9 System Error 2ms 532.0 KiB
#10 System Error 15ms 1.77 MiB
#11 System Error 52ms 3.02 MiB
#12 System Error 13ms 1.566 MiB
#13 System Error 31ms 3.445 MiB
#14 System Error 22ms 2.52 MiB
#15 System Error 2ms 532.0 KiB
#16 System Error 28ms 3.27 MiB
#17 System Error 28ms 3.27 MiB
#18 System Error 28ms 3.055 MiB
#19 System Error 142ms 18.879 MiB
#20 System Error 141ms 18.746 MiB
#21 System Error 136ms 18.996 MiB
#22 System Error 192ms 35.621 MiB
#23 System Error 189ms 18.426 MiB
#24 System Error 195ms 18.77 MiB
#25 System Error 162ms 18.996 MiB
#26 System Error 10ms 2.02 MiB

Code

#include<bits/stdc++.h>
using namespace std;
using ll=long long int;

void fastio(){ios::sync_with_stdio(false);cin.tie(0);}

vector<ll> take_inputs_0(ll n){vector<ll> v(n);for(int i=0;i<n;i++){cin>>v[i];}return v;}
vector<ll> take_inputs_1(ll n){vector<ll> v(n);for(int i=1;i<n;i++){cin>>v[i];}return v;}

void show(auto x){ cout<<x<<"\n";}
void show1(auto v){ ll n=(signed)v.size();for(int i=0;i<n;i++){cout<<v[i]<<" ";}cout<<"\n";}
void show2(auto v){ ll n=(signed)v.size();for(int i=0;i<n;i++){	cout<<v[i]<<"\n";}}

void sort_it(auto &v){ sort(v.begin(),v.end());}
void reverse_it(auto &v){ reverse(v.begin(),v.end());}
void do_both(auto &v){ sort(v.begin(),v.end());reverse(v.begin(),v.end());}

auto find_max(auto v){ sort(v.begin(),v.end());reverse(v.begin(),v.end());return v[0];}
auto find_min(auto v){ sort(v.begin(),v.end());return v[0];}

vector<ll> bitRep64(ll n){vector<ll> v;int i=0;while(i<=63){if(n&(1LL<<i)){v.push_back(1);}else{v.push_back(0);}i++;}return v;}
vector<ll> bitRep32(ll n){vector<ll> v;int i=0;while(i<=31){if(n&(1LL<<i)){v.push_back(1);}else{v.push_back(0);}i++;}return v;}
vector<string> all_subset(ll n){vector<string> v;for(int i=0;i<(1<<n);i++){string st="";for(int j=0;j<n;j++){if(i&(1<<j)){st+="1";}else{st+="0";}}v.push_back(st);}return v;}

vector<ll> prefix_sum_0(vector<ll> v){ll n=(signed)v.size();for(int i=1;i<n;i++){v[i]=v[i-1]+v[i];}return v;}
vector<ll> prefix_sum_1(vector<ll> v){ll n=(signed)v.size();vector<ll> summ(n+1);for(int i=1;i<=n;i++){summ[i]=summ[i-1]+v[i-1];}return summ;}

///////////////////////////////////

ll ans=0;
ll n;
ll done=0;

ll res=0;
void dfs(ll node,vector<ll> adj[],ll visited[],ll cnt[],ll restricted[])
{
	visited[node]=1;
	
	ll countt=0;
	
	for(auto x:adj[node])
	{
		if(!visited[x])
		{
			if(cnt[x]>=1){countt++;}
			cnt[node]+=cnt[x];
			//cnt[node]+=countt+1;
			dfs(x,adj,visited,cnt,restricted);
			countt+=cnt[x];
		}
	}
	
//	cout<<node<<" "<<cnt[node]<<"\n";
	cnt[node]+=countt;
	
	if(cnt[node]==0)
	{
		restricted[node]=1;
	}
}

ll dfs2(ll node,vector<ll> adj[],ll visited[],ll cnt[],ll restricted[])
{
	visited[node]=1;
	res=ans;
	for(auto x:adj[node])
	{
		if(!visited[x] && restricted[x]==0)
		{
			if(done<n){ans++;}
			done++;
			dfs2(x,adj,visited,cnt,restricted);
		}
	}
	if(done<n){ans++;}
	//show(ans);
}


int main()
{
	int t;cin>>t;
	
	while(t--)
	{
		n;cin>>n;
		
		vector<ll> v(n+1);
		
		ll visited[n+1];
		ll cnt[n+1];
		
        ll restricted[n+1];
        
        ans=0;
        done=0;
        res=0;
		for(int i=1;i<=n;i++)
		{
			cin>>v[i];
			visited[i]=0;
			restricted[i]=0;
			cnt[i]=v[i];
		}
		
		vector<ll> adj[n+1];
		
		for(int i=0;i<n-1;i++)
		{
			ll x,y;cin>>x>>y;
			
			adj[x].push_back(y);
			adj[y].push_back(x);
		}
		
		ll node=-1;
		
		
		for(int i=1;i<=n;i++)
		{
			if(v[i]==1)
			{
				node=i;
				break;
			}
		}
		
		if(node==-1)
		{
			show(0);
			continue;
		}
		
		//show(node);
		
		dfs(node,adj,visited,cnt,restricted);
		
		
		for(int i=1;i<=n;i++)
		{
			visited[i]=0;
			cnt[i]=0;
			
		}
		
		
		dfs2(node,adj,visited,cnt,restricted);
		
		
		cout<<res<<"\n";
		
		
		
	}
	
	return 0;
}

Information

Submit By
Type
Submission
Problem
P1078 Apple on Tree
Language
C++17 (G++ 13.2.0)
Submit At
2024-08-16 18:22:13
Judged At
2024-08-16 18:22:13
Judged By
Score
0
Total Time
195ms
Peak Memory
35.621 MiB