| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 155539 | Learner99 | Segway (COI19_segway) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
#define N 100006
int a[N][3];
#define ordered_set tree< pair < int , int > , null_type,less< pair < int , int > >, rb_tree_tag,tree_order_statistics_node_update>
ordered_set s[405];
int acc[405] = {0};
int cur[N];
int ans[N];
int32_t main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i][0]>>a[i][1]>>a[i][2];
}
int m;
cin>>m;
for(int i=0;i<m;i++)
{
int temp;
cin>>temp;
acc[temp] = 1;
}
set < pair < int , int > > rid;
for(int i=1;i<=n;i++)
{
rid.insert({0, i});
}
while(rid.size()!=0)
{
pair < int , int > t = *rid.begin();
rid.erase(t);
int tme = t.first;
int rider = t.second;
if( acc[cur[rider]] == 0)
{
cur[rider]++;
int x = 0;
if(cur[rider]<= 100)
{
ans[rider]+=a[rider][0];
x = a[rider][0];
}
else if(cur[rider]<=200)
{
ans[rider]+=a[rider][1];
x = a[rider][1];
}
else
{
ans[rider]+=a[rider][2];
x = a[rider][2];
}
if(cur[rider]<300)
{
rid.insert({tme+x, rider});
}
s[cur[rider]].insert({tme+x, rider});
}
else
{
int aage = s[cur[rider]].order_of_key({tme, 0});
aage%=20;
if(aage==0)
{
cur[rider]++;
int x = 0;
if(cur[rider]<= 100)
{
ans[rider]+=a[rider][0];
x = a[rider][0];
}
else if(cur[rider]<=200)
{
ans[rider]+=a[rider][1];
x = a[rider][1];
}
else
{
ans[rider]+=a[rider][2];
x = a[rider][2];
}
if(cur[rider]<300)
{
rid.insert({tme+x, rider});
}
s[cur[rider]].insert({tme+x, rider});
}
else
{
int y = 1;
int p = cur[rider];
for(int i=p+1;i<=min(300ll, p+aage);i++)
{
ans[rider]++;
cur[rider]++;
s[i].insert({tme+y, rider});
y++;
}
if(cur[rider] < 300)
rid.insert({tme+aage, rider});
}
}
}
for(int i=1;i<=n;i++)
{
cout<<ans[i]<<"\n";
}
cout<<endl;
return 0;
}