Submission #841875

#	Time	Username	Problem	Language	Result	Execution time	Memory
841875		vjudge1	Roadside Advertisements (NOI17_roadsideadverts)	C++17	7 / 100	1069 ms	5876 KiB

roadsideadverts

/*#pragma GCC optimize ("Ofast")
#pragma GCC optimize ("unroll-loops")
#pragma GCC target("avx,avx2,fma")*/
#include <iostream>
#include <vector>
using namespace std;
#define ll int
ll m[50010];
ll n,cnt,a1,a2,a3,a4,a5;
vector<pair<ll,ll>>g[50001];
void dfs(ll v,ll p,ll sum)
{
	if(m[v]==1)
	{
		cnt+=sum;sum=0;
	}
	//cout<<v<<" "<<sum<<"!!!\n";
	for(pair<ll,ll> it:g[v])
	{
		if(it.first!=p)
		{
			dfs(it.first,v,sum + it.second);
		}
	}
	//cout<<v<<" "<<sum<<"???\n";
}
int main()
{
	ios_base::sync_with_stdio();
    cin.tie(NULL);
    cout.tie(NULL);
    
    cin>>n;
    ll q,u,v,c;
    for(int i=1;i<n;i++)
    {
    	//cin>>u>>v>>c;
    	scanf("%d%d%d",&u,&v,&c);
    	g[u].push_back({v,c});
    	g[v].push_back({u,c});
	}
	cin>>q;
	while(q--)
	{
		scanf("%d%d%d%d%d",&a1,&a2,&a3,&a4,&a5);
		//cin>>a1>>a2>>a3>>a4>>a5;
		m[a2]=m[a3]=m[a4]=m[a5]=1;
		dfs(a1,-1,0);
		cout<<cnt<<"\n";
		
		m[a2]=m[a3]=m[a4]=m[a5]=0;
		cnt=0;
	}
}

Compilation message (stderr)

roadsideadverts.cpp: In function 'int main()':
roadsideadverts.cpp:38:11: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   38 |      scanf("%d%d%d",&u,&v,&c);
      |      ~~~~~^~~~~~~~~~~~~~~~~~~
roadsideadverts.cpp:45:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   45 |   scanf("%d%d%d%d%d",&a1,&a2,&a3,&a4,&a5);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 0 / 23
• Subtask 3: 0 / 40
• Subtask 4: 0 / 30