QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#728011#9529. Farm Management336699goWA 34ms11912kbC++174.9kb2024-11-09 14:22:312024-11-09 14:22:32

Judging History

你现在查看的是最新测评结果

  • [2024-11-09 14:22:32]
  • 评测
  • 测评结果:WA
  • 用时:34ms
  • 内存:11912kb
  • [2024-11-09 14:22:31]
  • 提交

answer

#include <bits/stdc++.h>
#define int long long
using namespace std;

int n, m, w, l, r;
int a[100005][3]{};
int seg[400005]{};

// idx번째 노드가 나타내는 구간이 st에서 en이고, l부터 r까지의 합을 구하고 싶다
int sum(int idx, int st, int en, int l, int r) {
    if(l > en || r < st) {
        return 0;
    } else if (l <= st && en <= r) {
        return seg[idx];
    } else {
        return sum(idx * 2, st, (st + en) / 2, l, r) + sum(idx * 2 + 1, (st + en) / 2 + 1, en, l, r);
    }
}

// idx번째 노드가 나타내는 구간이 st에서 en이고 i번재 원소를 diff만큼 변화시키고 싶다
void update(int idx, int st, int en, int i, int diff) {
    if (i >= st && i <= en) {
        seg[idx] += diff;
        if (st == en) {
            return;
        }
        if (i <= (st + en) / 2) {
            update(idx * 2, st, (st + en) / 2, i, diff);
        } else {
            update(idx * 2 + 1, (st + en) / 2 + 1, en, i, diff);
        }
    }
}

signed main() {
    ios::sync_with_stdio(0), cin.tie(0);
    cin >> n >> m;
    vector<vector<int>> vec(n, vector<int>(3));
    
    for (int i = 0; i < n; i++) {
        cin >> w >> l >> r;
        vec[i][0] = w;
        vec[i][1] = l;
        vec[i][2] = r;
    }

    sort(vec.begin(), vec.end(), [](const vector<int>& o1, const vector<int>& o2) {
        return o1[0] > o2[0];
    });

    vector<int> usedTime(n);
    vector<int> remainingTime(n);
    int total = 0;

    for (int i = 0; i < n; i++) {
        usedTime[i] = vec[i][2];
        total += vec[i][2];
    }

    for (int i = n - 1; total > m && i >= 0; i--) {
        usedTime[i] = max(vec[i][1], usedTime[i] - (total - m));
        total -= (vec[i][2] - usedTime[i]);
        remainingTime[i] = (vec[i][2] - usedTime[i]);
    }

    vector<int> prefixSum(n);
    vector<int> prefixSum_2(n);
    prefixSum[0] = remainingTime[0];
    prefixSum_2[0] = remainingTime[0] * vec[0][0];

    for (int i = 1; i < n; i++) {
        prefixSum[i] = prefixSum[i - 1] + remainingTime[i];
        prefixSum_2[i] = prefixSum_2[i - 1] + remainingTime[i] * vec[i][0];
    }

    int answer = 0;
    int lSum = 0;
    for (int i = n - 1; i > 0; i--) {
        answer += (vec[i][0] * vec[i][1]);
        lSum += vec[i][1];
    }
    answer += (vec[0][0] * (m - lSum));

    // cout << "usedTime:: ";
    // for (int i = 0; i < n; i++) {
    //     cout << usedTime[i] << " ";
    // }
    // cout << endl;
    // cout << "remainingTime:: ";
    // for (int i = 0; i < n; i++) {
    //     cout << remainingTime[i] << " ";
    // }
    // cout << endl;
    // cout << "prefixSum:: ";
    // for (int i = 0; i < n; i++) {
    //     cout << prefixSum[i] << " ";
    // }
    // cout << endl;
    // cout << "prefixSum_2:: ";
    // for (int i = 0; i < n; i++) {
    //     cout << prefixSum_2[i] << " ";
    // }
    // cout << endl;
    
    int haha = 0;

    for (int i = 0; i < n; i++) {
        haha += usedTime[i] * vec[i][0];
    }

    // cout << "haha :: " << haha << endl;

    for (int i = n - 1; i > 0; i--) {
        int time = usedTime[i];
        int addVal = -1 * usedTime[i] * vec[i][0];

        if (prefixSum[i - 1] <= 0) {
            continue;
        } else if (prefixSum[i - 1] <= time) {
            addVal += prefixSum_2[i - 1];
            time -= prefixSum[i - 1];

            if (time > 0) {
                int idx = lower_bound(prefixSum.begin() + i + 1, prefixSum.end(), time + prefixSum[i]) - prefixSum.begin();
                // cout << time << endl;
                // cout << "  idx :: " << idx << endl;
                if (idx == n) {
                    addVal += prefixSum_2[n - 1] - prefixSum_2[i];
                    addVal += (vec[i][0] * (time - (prefixSum[n - 1] - prefixSum[i])));
                } else if (prefixSum[idx] == time + prefixSum[i]) {
                    addVal += (prefixSum_2[idx] - prefixSum_2[i]);
                } else {
                    if (idx - 1 > i) { 
                        addVal += prefixSum_2[idx - 1];
                        time -= (prefixSum[idx - 1] - prefixSum[i]);
                    }
                    addVal += vec[idx][0] * time;
                }
            }
        } else {
            int idx = lower_bound(prefixSum.begin(), prefixSum.begin() + i, time) - prefixSum.begin();
            // cout << "idx:: " << idx << endl;
            if (prefixSum[idx] == time) {
                addVal += prefixSum_2[idx];
            } else {
                if (idx - 1 >= 0) { 
                    addVal += prefixSum_2[idx - 1];
                    time -= prefixSum[idx - 1];
                }
                addVal += vec[idx][0] * time;
            }
        }
        // cout << i << ":: " << haha + addVal << endl;
        answer = max(answer, haha + addVal);
    }

    cout << answer << endl;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3556kb

input:

5 17
2 3 4
6 1 5
8 2 4
4 3 3
7 5 5

output:

109

result:

ok single line: '109'

Test #2:

score: 0
Accepted
time: 0ms
memory: 3776kb

input:

12 62
503792 9 10
607358 1 3
600501 10 10
33249 4 4
774438 6 6
197692 3 6
495807 8 8
790225 5 9
77272 3 8
494819 4 9
894779 3 9
306279 5 6

output:

35204500

result:

ok single line: '35204500'

Test #3:

score: 0
Accepted
time: 0ms
memory: 3552kb

input:

15 32
835418 2 3
178262 1 3
527643 2 2
519710 1 1
774544 3 3
82312 1 1
808199 1 1
809396 1 3
255882 1 3
80467 1 3
874973 1 3
813965 1 2
198275 1 2
152356 1 3
802055 1 1

output:

22000255

result:

ok single line: '22000255'

Test #4:

score: 0
Accepted
time: 0ms
memory: 3844kb

input:

13 20
526447 1 1
807398 2 2
4167 1 2
944031 2 2
830685 2 2
394251 1 2
505011 1 2
968848 1 1
58170 1 3
32504 1 1
792273 3 3
196120 1 2
714507 1 1

output:

12878768

result:

ok single line: '12878768'

Test #5:

score: 0
Accepted
time: 0ms
memory: 3652kb

input:

13 32
582584 1 3
335440 3 3
971984 1 2
864169 1 2
528515 1 1
382399 1 2
459855 1 2
406909 2 3
66780 2 3
885118 3 3
434844 1 2
93331 1 3
502509 1 3

output:

22065034

result:

ok single line: '22065034'

Test #6:

score: 0
Accepted
time: 0ms
memory: 3552kb

input:

12 77
30244 1 7
518214 3 8
486001 8 9
152634 2 3
180255 3 4
791887 1 6
635820 2 9
881171 3 5
337905 3 8
683182 5 5
300786 3 6
339094 7 9

output:

50453764

result:

ok single line: '50453764'

Test #7:

score: 0
Accepted
time: 0ms
memory: 3552kb

input:

10 3923726
826284 215861 638800
471693 146746 886003
140800 532315 684546
673434 604071 814259
170671 299465 525449
104262 689547 855391
215333 591975 803421
795321 31606 984783
103838 361911 601318
145693 450227 686945

output:

1597735409747

result:

ok single line: '1597735409747'

Test #8:

score: 0
Accepted
time: 27ms
memory: 11792kb

input:

100000 16648414311
252800 55607 195981
157144 548469 789695
643048 2 2
907957 3 3
32532 231618 316365
194428 227513 762023
4231 393553 699179
898052 3 5
507551 3 5
747498 1 4
857939 9 9
440056 764429 796585
495571 117772 838593
4059 551203 870687
60877 597981 770178
593237 4 10
438147 218335 370780
...

output:

4148641232436282

result:

ok single line: '4148641232436282'

Test #9:

score: 0
Accepted
time: 34ms
memory: 11912kb

input:

100000 14997174848
996832 2 5
193379 411081 976749
339827 553492 639067
127364 131886 237768
615192 3 9
417612 599185 637969
812326 2 5
22605 80976 90557
350917 119223 755129
807528 7 8
165108 327186 793786
126946 530792 700713
246467 387234 557026
286888 485816 869662
398882 73798 458939
651741 3 9...

output:

3763949202789374

result:

ok single line: '3763949202789374'

Test #10:

score: -100
Wrong Answer
time: 27ms
memory: 11884kb

input:

99999 49959949282
541788 999954 999965
269794 8 9
446939 999985 999990
994146 7 13
976318 999916 999995
372141 999918 999934
398999 999949 999983
736560 4 7
753380 999933 999979
664693 12 14
336068 999904 999914
337406 11 11
153015 2 3
989608 1 15
61489 999921 999977
438155 999928 999955
374725 6 14...

output:

24951202749636060

result:

wrong answer 1st lines differ - expected: '24950941271114177', found: '24951202749636060'