Submission #795700

#TimeUsernameProblemLanguageResultExecution timeMemory
795700alexander707070Packing Biscuits (IOI20_biscuits)C++14
0 / 100
1095 ms162160 KiB
#include<bits/stdc++.h>
using namespace std;

const long long mod=1e9+7;

long long x,sum;
int k;
vector<long long> br,toadd;

unordered_map<long long,long long> dp;

long long brute(int bit,long long total,vector<long long> br){
    if(bit==-1)return 1;
    
    long long hesh=(total%mod)*100+bit;
    if(dp[hesh]>0)return dp[hesh];

    long long curr=br[bit];
    total-=curr*(1LL<<bit); br[bit]=0;
    long long res=brute(bit-1,total,br);
    total+=curr*(1LL<<bit); br[bit]=curr;

    curr=x;
    if(total>=x*(1LL<<bit)){
        total-=x*(1LL<<bit);
        
        int pos=bit;
        for(int i=bit;i>=0;i--){
            if(br[i]>=curr){
                br[i]-=curr; 
                curr=0; pos=i; break;
            }else{
                curr-=br[i]; 
                br[i]=0; curr*=2;
            }
        }

        total-=br[bit]*(1LL<<bit); br[bit]=0;
        res+=brute(bit-1,total,br);
    }

    dp[hesh]=res;
    return res;
}

long long count_tastiness(long long X,vector<long long> A){
    k=int(A.size()); x=X; sum=0; dp.clear();
    br.resize(62); toadd.resize(62);

    for(int i=0;i<k;i++){
        br[i]=A[i]; 
        sum+=br[i]*(1LL<<i);
    }

    return brute(61,sum,br);
}

/*
int main(){
    cout<<count_tastiness(2, {2, 1, 2})<<"\n";
    cout<<count_tastiness(2, {2, 1, 2})<<"\n";
}
*/

Compilation message (stderr)

biscuits.cpp: In function 'long long int brute(int, long long int, std::vector<long long int>)':
biscuits.cpp:27:13: warning: variable 'pos' set but not used [-Wunused-but-set-variable]
   27 |         int pos=bit;
      |             ^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...