Submission #1072919

#	Time	Username	Problem	Language	Result	Execution time	Memory
1072919		Ignut	비스킷 담기 (IOI20_biscuits)	C++17	12 / 100	1 ms	604 KiB

biscuits

// Ignut

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

ll count_tastiness(ll x, vector<ll> a) {
    int k = a.size();
    for (int i = 0; i < k; i ++) {
        if (a[i] > 2) {
            if (i == a.size() - 1) {
                a.push_back(0);
                k ++;
            }
            int left = (a[i] % 2 == 1 ? 1 : 2);
            a[i + 1] += (a[i] - left) / 2;
            a[i] = left;
        }
    }
    a.push_back(0);
    k ++;

    // for (int val : a) cerr << val << ' ';
    // cout << '\n';

    ll dp[k + 1][4] = {};
    dp[0][0] = 1;
    for (int i = 0; i < k; i ++) {
        for (int prev = 0; prev <= 3; prev ++) {
            ll val = a[i] + prev / 2;
            // take 1
            if (val >= 1)
                dp[i + 1][val - 1] += dp[i][prev];
            // take 0
            dp[i + 1][val] += dp[i][prev];
        }
    }

    // for (int i = 0; i <= k; i ++) {
    //     for (int v = 0; v <= 3; v ++) {
    //         cout << dp[i][v] << ' ';
    //     }
    //     cout << '\n';
    // }

    ll res = 0;
    for (int v = 0; v <= 3; v ++) res += dp[k][v];
    return res;
}

Compilation message (stderr)

biscuits.cpp: In function 'll count_tastiness(ll, std::vector<long long int>)':
biscuits.cpp:12:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 |             if (i == a.size() - 1) {
      |                 ~~^~~~~~~~~~~~~~~

• Subtask 1: 0 / 9
• Subtask 2: 12 / 12
• Subtask 3: 0 / 21
• Subtask 4: 0 / 35
• Subtask 5: 0 / 23