Submission #640578

#	Time	Username	Problem	Language	Result	Execution time	Memory
640578		Tuanlinh123	Paprike (COI18_paprike)	C++17	13 / 100	39 ms	12916 KiB

paprike

#include<bits/stdc++.h>
#define ll long long
#define ld long double
#define pll pair<ll,ll>
#define mp make_pair
#define pb push_back
#define fi first
#define se second

using namespace std;

#define LOCALIO "C:/Users/admin/Documents/Code/freopen/"

vector <ll> A[100005];
ll a[100005], ans, n, k, num;
bool used[100005];

void dfs(ll s)
{
    used[s]=1;
    for (ll i=0; i<A[s].size(); i++)
    {
        ll v=A[s][i];
        if (used[v])
            continue;
        if (a[v]+num<=k)
        {
            num+=a[v];
            dfs(v);
        }
    }
    for (ll i=0; i<A[s].size(); i++)
    {
        ll v=A[s][i];
        if (used[v])
            continue;
        num=a[v];
        ans++;
        dfs(v);
    }
}


int main()
{
    #ifdef LOCAL
        freopen( LOCALIO "input.txt","r",stdin) ;
        freopen( LOCALIO "output.txt","w",stdout) ;
    #endif

    ios_base::sync_with_stdio(NULL); cin.tie(nullptr); cout.tie(nullptr);
//	freopen("FIBONACCI.inp","r",stdin);
//	freopen("FIBONACCI.out","w",stdout);
    cin >> n >> k;
    for (ll i=1; i<=n; i++)
        cin >> a[i];
    for (ll i=1; i<n; i++)
    {
        ll x, y; cin >> x >> y;
        A[x].pb(y); A[y].pb(x);
    }
    num=a[1];
    dfs(1);
    cout << ans;
}

Compilation message (stderr)

paprike.cpp: In function 'void dfs(long long int)':
paprike.cpp:21:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   21 |     for (ll i=0; i<A[s].size(); i++)
      |                  ~^~~~~~~~~~~~
paprike.cpp:32:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |     for (ll i=0; i<A[s].size(); i++)
      |                  ~^~~~~~~~~~~~

• Subtask 1: 0 / 11
• Subtask 2: 13 / 13
• Subtask 3: 0 / 27
• Subtask 4: 0 / 49