Submission #644760

#	Time	Username	Problem	Language	Result	Execution time	Memory
644760		Vu_CG_Coder	Cat in a tree (BOI17_catinatree)	C++14	51 / 100	461 ms	33492 KiB

catinatree

/* [Author : Hoang Duy Vu] */
#include <bits/stdc++.h>
#define All(x) (x).begin(),(x).end()
#define ll long long
#define C make_pair
#define fi first
#define se second
#define two second.first
#define thr second.second
#define TASK "txt"
using namespace std;
template<typename T> bool maximize(T &res, const T &val) {
    if (res < val) { res = val; return true; } return false; }
template<typename T> bool minimize(T &res, const T &val) {
    if (res > val) { res = val; return true; } return false; }
 
typedef pair<int,int> ii;
typedef pair<int,ii> iii;
const int LOG = 20;
const int INF = 1e9 + 7;
const ll LNF = 1e18 + 7;
const int mod = 1e9 + 7;
const int N = 200100;
 
vector <int> a[N];
int dp[1600][3][1600] = {0};
int t[N] = {0};
int high[N] = {0};
int n , k;
 
void input()
{
    cin >> n >> k;
    for (int i = 1 ; i < n ; i++)
    {
        int x;
        cin >> x;
        a[x].push_back(i);
        a[i].push_back(x);
    }
}
 
void dfs(int u ,int par)
{
    memset(dp[u],0,sizeof(dp[u]));
    int cnt = 0;
 
    for (int v : a[u])
    if (v != par)
    {
        dfs(v,u);
        int New = (cnt + 1)&1;
        memset(dp[u][New],0,sizeof(dp[New][0]));
        maximize(high[u],high[v] + 1);
 
        for (int i = 0 ; i <= min(high[u],k) ; i++)
        {
            dp[u][New][i] = dp[u][cnt][i];
 
            for (int j = max(i - 1,0) ; j <= min(high[v],k) ; j++)
            {
                int x = j + 1;
                maximize(dp[u][New][i],dp[u][cnt][max(i,k - x)] + dp[v][t[v]][j]);
            }
 
 
        }

        cnt = New;
    }
 
    t[u] = cnt;
 
    maximize(dp[u][t[u]][0],dp[u][t[u]][k] + 1);
}
 
void solve()
{
    dfs(0,0);
 
    int res = 0;
    for (int i = 0 ; i <= high[0] ; i++) maximize(res,dp[0][t[0]][i]);
 
    cout << res;
}
 
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(NULL); cout.tie(NULL);
    if(fopen(TASK".inp", "r")){
    freopen(TASK".inp","r",stdin);
    freopen(TASK".out","w",stdout);
    }
 
    int tk;
    tk = 1;
   //cin >> tk;
    while (tk--)
    {
        input();
        solve();
    }
    return 0;
}

Compilation message (stderr)

catinatree.cpp: In function 'int main()':
catinatree.cpp:92:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   92 |     freopen(TASK".inp","r",stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
catinatree.cpp:93:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   93 |     freopen(TASK".out","w",stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~

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