Submission #986412

#	Time	Username	Problem	Language	Result	Execution time	Memory
986412		Neco_arc	Cat in a tree (BOI17_catinatree)	C++17	0 / 100	2 ms	9816 KiB

catinatree

#include <bits/stdc++.h>
//#include <bits/debug.hpp>

#define ll long long
#define all(x) x.begin(), x.end()
#define Neco "Maximum Weight Subset"
#define resp(x) sort(all(x)), x.resize(unique(all(x)) - x.begin())
#define getbit(x,i) ((x >> i)&1)
#define _left id * 2, l, mid
#define _right id * 2 + 1, mid + 1, r
#define cntbit(x) __builtin_popcountll(x)
#define fi(i, a, b) for(int i = a; i <= b; i++)
#define fid(i, a, b) for(int i = a; i >= b; i--)
#define maxn (int) 2e5 + 7

using namespace std;

const ll mod = 1e9 + 7; //972663749
const ll base = 911382323;

int n, k;
int a[maxn];
vector<int> edges[maxn];
vector<int> dp[maxn];

vector<int> cb(const vector<int> &a, const vector<int> &b) {
    vector<int> ans(a.size() + b.size() - 1, -1e9);
    fi(i, 0, a.size() - 1) fi(j, 0, b.size() - 1) {
        if(i + j >= k) ans[min(i, j)] = max(ans[min(i, j)], a[i] + b[j]);
    }
    fi(i, 0, b.size() - 1) ans[i] = max(ans[i], b[i]);
    fi(i, 0, a.size() - 1) ans[i] = max(ans[i], a[i]);
    return ans;
}

void dfs(int u, int par) {
    dp[u] = {a[u]};
    for(int v : edges[u]) if(v != par) {
        dfs(v, u);
        dp[u] = cb(dp[u], dp[v]);
    }

//    if(u == 3) cout << dp[u] << '\n';
    dp[u].insert(dp[u].begin(), -1e9);
}

void solve()
{

    cin >> n >> k;
    fi(i, 2, n) {
        int x; cin >> x, ++x;
        edges[x].push_back(i);
    }

    dfs(1, 0);
    cout << *max_element(all(dp[1]));


}


int main()
{

    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    if(fopen(Neco".inp", "r")) {
        freopen(Neco".inp", "r", stdin);
        freopen(Neco".out", "w", stdout);
    }


    int nTest = 1;
//    cin >> nTest;


    while(nTest--)
    {
        solve();
    }


    return 0;
}

Compilation message (stderr)

catinatree.cpp: In function 'std::vector<int> cb(const std::vector<int>&, const std::vector<int>&)':
catinatree.cpp:12:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 | #define fi(i, a, b) for(int i = a; i <= b; i++)
......
   28 |     fi(i, 0, a.size() - 1) fi(j, 0, b.size() - 1) {
      |        ~~~~~~~~~~~~~~~~~~             
catinatree.cpp:28:5: note: in expansion of macro 'fi'
   28 |     fi(i, 0, a.size() - 1) fi(j, 0, b.size() - 1) {
      |     ^~
catinatree.cpp:12:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 | #define fi(i, a, b) for(int i = a; i <= b; i++)
......
   28 |     fi(i, 0, a.size() - 1) fi(j, 0, b.size() - 1) {
      |                               ~~~~~~~~~~~~~~~~~~
catinatree.cpp:28:28: note: in expansion of macro 'fi'
   28 |     fi(i, 0, a.size() - 1) fi(j, 0, b.size() - 1) {
      |                            ^~
catinatree.cpp:12:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 | #define fi(i, a, b) for(int i = a; i <= b; i++)
......
   31 |     fi(i, 0, b.size() - 1) ans[i] = max(ans[i], b[i]);
      |        ~~~~~~~~~~~~~~~~~~             
catinatree.cpp:31:5: note: in expansion of macro 'fi'
   31 |     fi(i, 0, b.size() - 1) ans[i] = max(ans[i], b[i]);
      |     ^~
catinatree.cpp:12:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 | #define fi(i, a, b) for(int i = a; i <= b; i++)
......
   32 |     fi(i, 0, a.size() - 1) ans[i] = max(ans[i], a[i]);
      |        ~~~~~~~~~~~~~~~~~~             
catinatree.cpp:32:5: note: in expansion of macro 'fi'
   32 |     fi(i, 0, a.size() - 1) ans[i] = max(ans[i], a[i]);
      |     ^~
catinatree.cpp: In function 'int main()':
catinatree.cpp:71:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   71 |         freopen(Neco".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
catinatree.cpp:72:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   72 |         freopen(Neco".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 11
• Subtask 2: 0 / 40
• Subtask 3: 0 / 49