Submission #1215069

#	Time	Username	Problem	Language	Result	Execution time	Memory
1215069		sasde	City (BOI06_city)	C++20	100 / 100	50 ms	580 KiB

city

#include<bits/stdc++.h>
using namespace std;
bool M1;
#define task "aws"
#define sz(a) (int)a.size()
#define all(x) x.begin(),x.end()
#define ii pair<long long,int>
#define iii pair<int,ii>
#define iv pair<ii,ii>
#define se second
#define fi first
#define ffi fi.fi
#define sfi se.fi
#define sse se.se
#define fse fi.se
#define lt(i, c, d) for(int i = c; i <= d; ++i)
#define fl(i, c, d) for(int i = d; i >= c; --i)
#define pb push_back
#define emb emplace_back
#define emf emplace_front
#define em emplace
#define look_memory cerr<<abs(&M2-&M1)/1024.0/1024<<'\n'
#define look_time   cerr << "TIME : " << clock() * 0.001 << "s" <<'\n'
const int N=1e6+5,lg=30,mod=1e9+7;
mt19937 rd(chrono::steady_clock::now().time_since_epoch().count());
int Rand(int u,int v){
 return u+rd()%(v-u+1);
}
int dx[]={1,0,-1,0,1,1,-1,-1};
int dy[]={0,-1,0,1,1,-1,1,-1};
long long persons,distes,floors,cost[N],cost1[N];
bool check(long long mid){
    long long res=0;
    for(long long i=4,d=0,tang=floors,sum=0;;i+=4,++d){
        while(tang>0&&distes*d+cost1[tang]>mid)--tang;
        // if(tang==0)break;
        // cout <<distes*d+cost1[tang]<<" "<<tang<<" "<<res<<'\n';
        res+=i*tang;
        sum+=i;
        res=min(res,persons);
        if(sum>=persons)break;
    }
    return res>=persons;
}
bool M2;
void solve(){
    cin >> persons >> distes >> floors;
    for(int i=1;i<=floors;++i){
        cin >> cost1[i];
        cost[i]=cost1[i];
        cost[i]+=cost[i-1];
    }
    long long r=0,l=1e18,res=0;
    while(r<=l){
        long long mid=(r+l)>>1;
        if(check(mid)){
            l=mid-1;
            res=mid;
        }
        else r=mid+1;
        // cout <<r<<" "<<l<<'\n';
    }
    long long ans=0,n=persons;
    for(long long i=4,d=0,tang=floors,sum=0;;i+=4,++d){
        while(tang>0&&distes*d+cost1[tang]>res-1)--tang;
        ans+=i*(cost[tang]+distes*d*tang);
        n-=i*tang;
        // cout << tang<<" "<<distes*d+cost1[tang]<<'\n';
        sum+=i;
        if(sum>=persons)break;
    }
    cout << ans+n*res;;
}
main()
{
  srand(time(0));
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    if(fopen(task".inp","r")){
      freopen(task".inp","r",stdin);
      freopen(task".out","w",stdout);
    }
    int t=1;
 //   cin >> t;
while(t--){
    solve();
}
look_memory;
look_time;
}

Compilation message (stderr)

city.cpp:74:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   74 | main()
      | ^~~~
city.cpp: In function 'int main()':
city.cpp:81:14: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   81 |       freopen(task".inp","r",stdin);
      |       ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
city.cpp:82:14: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   82 |       freopen(task".out","w",stdout);
      |       ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 100 / 100