QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#263100 | #7744. Elevator | ucup-team1134# | AC ✓ | 62ms | 6860kb | C++17 | 7.1kb | 2023-11-24 15:17:39 | 2023-11-24 15:17:41 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return true; } return false; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return true; } return false; }
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define mp make_pair
#define si(x) int(x.size())
const int mod=998244353,MAX=300005,INF=1<<30;
ll solve1(vector<pair<ll,ll>> A,vector<pair<ll,ll>> B,ll K){
ll ans=0;
sort(all(A));
sort(all(B));
reverse(all(A));
reverse(all(B));
int i=0,j=0;
ll now=0;
while(i<si(A)||j<si(B)){
if(i==si(A)){
if(now>=K-1) now=0;
if(now){
ll can=min((K-now)/2,B[j].se);
now+=can*2;
B[j].se-=can;
if(now>=K-1) now=0;
if(B[j].se==0) j++;
}else{
ll can=min((K-now)/2,B[j].se);
ll tk=B[j].se/can;
if(tk>=2){
ans+=B[j].fi*tk;
B[j].se-=can*tk;
if(B[j].se==0) j++;
}else{
if(now==0) ans+=B[j].fi;
ll can=min((K-now)/2,B[j].se);
now+=can*2;
B[j].se-=can;
if(now>=K-1) now=0;
if(B[j].se==0) j++;
}
}
}else if(j==si(B)){
if(now){
ll can=min((K-now),A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}else{
ll can=min(K-now,A[i].se);
ll tk=A[i].se/can;
if(tk>=2){
ans+=A[i].fi*tk;
A[i].se-=can*tk;
if(A[i].se==0) i++;
}else{
if(now==0) ans+=A[i].fi;
ll can=min(K-now,A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}
}
}else if(now==K-1){
if(now==0) ans+=A[i].fi;
ll can=min(K-now,A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}else if(A[i].fi<=B[j].fi){
if(now>=K-1) now=0;
if(now){
ll can=min((K-now)/2,B[j].se);
now+=can*2;
B[j].se-=can;
if(now>=K) now=0;
if(B[j].se==0) j++;
}else{
ll can=min((K-now)/2,B[j].se);
ll tk=B[j].se/can;
if(tk>=2){
ans+=B[j].fi*tk;
B[j].se-=can*tk;
if(B[j].se==0) j++;
}else{
if(now==0) ans+=B[j].fi;
ll can=min((K-now)/2,B[j].se);
now+=can*2;
B[j].se-=can;
if(now>=K) now=0;
if(B[j].se==0) j++;
}
}
}else{
if(now){
ll can=min((K-now),A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}else{
ll can=min(K-now,A[i].se);
ll tk=A[i].se/can;
if(tk>=2){
ans+=A[i].fi*tk;
A[i].se-=can*tk;
if(A[i].se==0) i++;
}else{
if(now==0) ans+=A[i].fi;
ll can=min(K-now,A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}
}
}
//cout<<i<<" "<<j<<" "<<now<<" "<<ans<<endl;
}
return ans;
}
ll solve2(vector<pair<ll,ll>> A,vector<pair<ll,ll>> B,ll K){
ll ans=0;
sort(all(A));
sort(all(B));
reverse(all(A));
reverse(all(B));
int i=0,j=0;
ll now=0;
while(i<si(A)||j<si(B)){
if(i==si(A)){
if(now==0) ans+=B[j].fi;
ll can=min((K-now)/2,B[j].se);
now+=can*2;
B[j].se-=can;
if(now>=K-1) now=0;
if(B[j].se==0) j++;
}else if(j==si(B)){
if(now==0) ans+=A[i].fi;
ll can=min(K-now,A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}else if(now==K-1){
if(now==0) ans+=A[i].fi;
ll can=min(K-now,A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}else if(A[i].fi<=B[j].fi){
if(now==0) ans+=B[j].fi;
ll can=min((K-now)/2,B[j].se);
now+=can*2;
B[j].se-=can;
if(now>=K) now=0;
if(B[j].se==0) j++;
}else{
if(now==0) ans+=A[i].fi;
ll can=min(K-now,A[i].se);
now+=can;
A[i].se-=can;
if(now>=K) now=0;
if(A[i].se==0) i++;
}
//cout<<i<<" "<<j<<" "<<now<<" "<<ans<<endl;
}
return ans;
}
int main(){
std::ifstream in("text.txt");
std::cin.rdbuf(in.rdbuf());
cin.tie(0);
ios::sync_with_stdio(false);
/*
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
ll q=0;
while(1){
int N=rng()%100+1;
ll K=rng()%100+1;
K*=2;
vector<pair<ll,ll>> A,B;
for(int q=0;q<N;q++){
ll a,b,c;
b=rng()%2+1;
a=rng()%1000+1;
c=rng()%1000+1;
if(b==1) A.push_back(mp(c,a));
else B.push_back(mp(c,a));
}
ll x=solve1(A,B,K),y=solve2(A,B,K);
if(x!=y){
cout<<x<<" "<<y<<endl;
cout<<K<<endl;
for(auto [a,b]:A) cout<<'A'<<" "<<a<<" "<<b<<endl;
for(auto [a,b]:B) cout<<'B'<<" "<<a<<" "<<b<<endl;
return 0;
}
q++;
if(q%10000==0) cout<<q<<endl;
}
*/
int Q;cin>>Q;
while(Q--){
ll N,K;cin>>N>>K;
vector<pair<ll,ll>> A,B;
ll ans=0;
for(int i=0;i<N;i++){
ll a,b,c;cin>>a>>b>>c;
if(b==1) A.push_back(mp(c,a));
else B.push_back(mp(c,a));
}
cout<<solve1(A,B,K)<<"\n";
}
}
这程序好像有点Bug,我给组数据试试?
详细
Test #1:
score: 100
Accepted
time: 1ms
memory: 3504kb
input:
2 4 6 1 1 8 7 2 5 1 1 7 3 2 6 8 1200000 100000 1 100000 100000 1 12345 100000 2 100000 100000 2 12345 100000 1 100000 100000 1 12345 100000 2 100000 100000 2 12345
output:
24 100000
result:
ok 2 lines
Test #2:
score: 0
Accepted
time: 38ms
memory: 6220kb
input:
5501 8 104 5 2 3 6 2 4 5 2 3 6 2 9 8 2 4 2 1 3 7 2 4 8 2 8 1 290 3 1 1 12 12 6 1 2 1 2 2 1 1 2 1 2 4 6 1 1 1 2 5 6 1 4 4 1 4 6 2 4 6 2 5 4 2 5 4 1 4 5 334 1 1 4 1 2 3 4 2 1 5 1 1 2 1 2 13 218 5 2 3 5 1 4 1 2 1 1 2 5 3 2 2 1 1 3 4 2 2 1 2 5 2 2 1 2 1 5 3 2 1 5 2 1 1 1 4 10 260 7 2 1 5 1 1 5 2 4 6 1 6...
output:
9 1 23 4 5 7 1 3 9 6 1 10 4 9 17 6 4 1 8 5 5 7 1 3 23 6 3 3 2 2 2 3 8 1 5 6 9 11 147 7 10 2 7 7 8 6 5 5 1 7 3 5 10 7 7 10 8 1 4 2 3 9 1 5 2 9 1 6 7 7 6 10 18 8 10 4 10 9 2 8 3 5 9 3 6 5 3 2 6 1 3 2 2 1 6 9 6 3 4 8 9 9 2 6 1 2 6 7 5 2 5 21 8 1 2 3 4 9 3 4 6 5 9 6 1 7 3 7 3 2 2 8 7 3 5 9 7 10 7 3 2 4 ...
result:
ok 5501 lines
Test #3:
score: 0
Accepted
time: 62ms
memory: 6860kb
input:
3 100000 8 8 2 6 7 2 8 5 1 10 7 2 7 4 1 7 6 2 9 7 1 2 3 1 1 5 1 10 5 1 1 3 2 10 5 2 5 7 1 6 5 2 7 7 2 1 5 1 6 2 2 10 8 1 7 7 1 1 10 2 6 9 1 1 8 1 4 10 1 4 6 1 1 4 1 4 1 2 9 3 2 6 6 1 3 4 1 7 9 2 5 8 2 4 7 2 1 2 2 10 6 2 9 3 1 3 2 2 3 2 2 4 7 2 4 8 1 1 5 2 3 5 2 7 7 2 9 1 1 4 5 1 2 2 1 9 3 2 1 9 2 2 ...
output:
568601 4708350936 93900404329230
result:
ok 3 lines
Test #4:
score: 0
Accepted
time: 54ms
memory: 6504kb
input:
3 100000 174 90429 1 64237 30851 1 44358 63571 2 89174 20391 2 28747 13561 2 88689 81508 2 28383 85733 1 5777 37524 2 34730 10588 2 88519 61493 2 83682 55885 1 65270 17669 2 63015 85387 1 64757 91420 2 74020 9067 2 91750 20809 1 52771 36478 2 17941 62064 1 36433 72912 2 6100 53923 2 73971 65825 2 39...
output:
2156004461915 884586357480 59034901233
result:
ok 3 lines
Test #5:
score: 0
Accepted
time: 44ms
memory: 6752kb
input:
3 100000 4418822 19907 1 13081 59730 2 93611 35050 1 5867 87777 1 21890 18374 1 82418 79526 2 76106 33510 2 45826 75573 1 42240 35449 1 98727 80720 2 65290 32021 2 51348 52399 2 97828 75498 2 51728 89905 1 22825 2855 1 26204 11427 1 99583 55530 2 37895 22256 2 92230 19533 1 9142 16138 1 54018 53102 ...
output:
84699074 270668 100000
result:
ok 3 lines
Extra Test:
score: 0
Extra Test Passed