Language: C++

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int a[5005];
ll p[5005];
bool vis[5005];
int tot=0;
vector<int>vec[5005];
map<int,ll>mp;
int main()
{
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++) cin>>p[i];
    for(int i=1;i<=n;i++)
    {
        if(vis[i]) continue;
        tot++;
        int t=i;
        ll ans=0;
        while(vis[t]==0)
        {
            vis[t]=1;
            vec[tot].push_back(t);
            ans+=p[t];
            t=a[t];
        }
        mp[tot]=ans;
    }
    ll res=p[1];
    for(int i=1;i<=tot;i++)
    {
        for(int j=0;j<vec[i].size();j++)
        {
            int t=vec[i][j];
            int step=0;
            ll ans=0;
            do
            {
                step++;
                t=a[t];
                ans+=p[t];
                if(mp[i]>0)
                {
                    res=max(res,ans+mp[i]*(k-step)/vec[i].size());
                }
                res=max(res,ans);
            }while(t!=vec[i][j]&&step<k);
        }
    }
    cout<<res;
    return 0;
}


Compiler output:
a.cc: In function 'int main()':
a.cc:46:28: error: no matching function for call to 'max(long long int&, long long unsigned int)'
   46 |                     res=max(res,ans+mp[i]*(k-step)/vec[i].size());
      |                         ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
In file included from /usr/include/c++/14/algorithm:60,
                 from /usr/include/x86_64-linux-gnu/c++/14/bits/stdc++.h:51,
                 from a.cc:1:
/usr/include/c++/14/bits/stl_algobase.h:257:5: note: candidate: 'template<class _Tp> constexpr const _Tp& std::max(const _Tp&, const _Tp&)'
  257 |     max(const _Tp& __a, const _Tp& __b)
      |     ^~~
/usr/include/c++/14/bits/stl_algobase.h:257:5: note:   template argument deduction/substitution failed:
a.cc:46:28: note:   deduced conflicting types for parameter 'const _Tp' ('long long int' and 'long long unsigned int')
   46 |                     res=max(res,ans+mp[i]*(k-step)/vec[i].size());
      |                         ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
/usr/include/c++/14/bits/stl_algobase.h:303:5: note: candidate: 'template<class _Tp, class _Compare> constexpr const _Tp& std::max(const _Tp&, const _Tp&, _Compare)'
  303 |     max(const _Tp& __a, const _Tp& __b, _Compare __comp)
      |     ^~~
/usr/include/c++/14/bits/stl_algobase.h:303:5: note:   candidate expects 3 arguments, 2 provided
In file included from /usr/include/c++/14/algorithm:61:
/usr/include/c++/14/bits/stl_algo.h:5706:5: note: candidate: 'template<class _Tp> constexpr _Tp std::max(initializer_list<_Tp>)'
 5706 |     max(initializer_list<_Tp> __l)
      |     ^~~
/usr/include/c++/14/bits/stl_algo.h:5706:5: note:   candidate expects 1 argument, 2 provided
/usr/include/c++/14/bits/stl_algo.h:5716:5: note: candidate: 'template<class _Tp, class _Compare> constexpr _Tp std::max(initializer_list<_Tp>, _Compare)'
 5716 |     max(initializer_list<_Tp> __l, _Compare __comp)
      |     ^~~
/usr/include/c++/14/bits/stl_algo.h:5716:5: note:   template argument deduction/substitution failed:
a.cc:46:28: note:   mismatched types 'std::initializer_list<_Tp>' and 'long long int'
   46 |                     res=max(res,ans+mp[i]*(k-step)/vec[i].size());
      |                         ~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~