Submission #963141

#	Time	Username	Problem	Language	Result	Execution time	Memory
963141		vjudge1	Fibonacci representations (CEOI18_fib)	C++14	15 / 100	4073 ms	5316 KiB

fib

#include<bits/stdc++.h>

using namespace std;

#define int long long

const int N=1e5+10, mod=1e9+7;
int n, a[N], f[N][2];

int32_t main(){
   ios_base::sync_with_stdio(false);
   cin.tie(nullptr);
   cin >> n;
   for (int i=1; i<=n; ++i) cin >> a[i];
   map<int, int> mp;
   for (int i=1; i<=n; ++i){
      ++mp[a[i]];
      for (auto it=mp.begin(); it!=mp.end();){
         if (it!=mp.begin() && prev(it)->first+1==it->first){
            int x=it->first;
            int d=min(prev(it)->second, it->second);
            mp[x+1]+=d;
            prev(it)->second-=d;
            if (prev(it)->second==0) mp.erase(prev(it));
            it->second-=d;
            if (it->second==0) it=mp.erase(it);
            while (it!=mp.begin() && prev(it)->first>=x-1) --it;
         }else if (it->second>=2){
            int x=it->first;
            int d=it->second/2;
            if (x==1){
               if (it->second==2){
                  ++mp[2];
                  it=mp.erase(it);
               }else{
                  mp[3]+=it->second/3;
                  it->second%=3;
                  if (it->second==0) it=mp.erase(it);
                  else ++it;
               }
            }else if (x==2){
               int sum=it->second*2;
               mp[3]+=sum/3;
               if (sum%3) mp[1]+=sum%3;
               mp.erase(it);
               it=mp.begin();
            }else{
               mp[x-2]+=d;
               mp[x+1]+=d;
               it->second-=d;
               if (it->second==0) mp.erase(it);
               while (it!=mp.begin() && prev(it)->first>=x-1) --it;
            }
         }else ++it;
      }
      vector<int> v{-1};
      for (auto &j:mp) v.push_back(j.first);
      f[1][0]=(v[1]+1)/2;
      f[1][1]=f[1][0]-1;
      for (int j=2; j<(int)v.size(); ++j){
         if ((v[j]&1)==(v[j-1]&1)){
            f[j][0]=(f[j-1][0]*(v[j]-v[j-1])/2+f[j-1][1]+mod)%mod;
            f[j][1]=(f[j][0]-f[j-1][0]+mod)%mod;
         }else{
            f[j][0]=f[j-1][0]*(v[j]-v[j-1]+1)/2%mod;
            f[j][1]=(f[j][0]-f[j-1][0]+mod)%mod;
         }
      }
      cout << f[(int)v.size()-1][0] << '\n';
   }
   return 0;
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 20
• Subtask 3: 15 / 15
• Subtask 4: 0 / 10
• Subtask 5: 0 / 15
• Subtask 6: 0 / 35