| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1051217 | nightfal | 비스킷 담기 (IOI20_biscuits) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <cassert>
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
#define ll unsigned long long
template <typename T> void print(T elem) {cout << elem << " ";}
template <typename T> void print(vector<T> &v) {for(auto elem: v) print(elem); cout << endl;}
void arrange(ll x, vector<ll> &a) {
int k=a.size();
for(int i=0; i<k-1; i++)
if (a[i] > x+1) {
ll excess = (a[i]-x)/2;
a[i] -= 2*excess;
a[i+1] += excess;
}
return;
}
int possible(ll n, ll x, vector<ll> a) {
int k = a.size();
// cout << "n:" << n << endl;
ll m = n;
for(int i=0; i<k && m>0; i++) {
bool one = m%2; m >>= 1;
if (a[i] > x+1) {
ll excess = (a[i]-x)/2;
a[i] -= 2*excess;
a[i+1] += excess;
}
if (one) {if (a[i]<x) return 0;}
else if (i<k-1) {a[i+1] += a[i]/2; a[i]=0;}
// cout << "i:" << i << ", a: "; print(a);
}
if (m>0) return 0;
else return 1;
}
ll subtask2(vector<ll> a) {
int k=a.size();
ll prev = 0, curr = 1;
for(int i=0; i<k; i++) {
switch(a[i]) {
case 0: curr += prev; prev = 0; break;
case 1: curr *=2; break;
case 2: prev += curr; curr *=2;
}
// cout << "i:" << i << " prev:" << prev << " curr:" << curr << endl;
}
return curr;
}
ll subtask1(ll x, vector<ll> a) {
ll cnt = 0;
for(int i=0; i<=100'000; i++) {
cnt += possible(i,x,a);
}
return cnt;
}
ll count_tastiness(ll x, std::vector<ll> a) {
a.resize(60,0);
arrange(x,a);
if (x==1) return subtask2(a);
return subtask1(x, a);
}