/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 540.0 KiB
#2 Accepted 20ms 540.0 KiB
#3 Accepted 32ms 540.0 KiB
#4 Accepted 31ms 588.0 KiB
#5 Accepted 30ms 540.0 KiB
#6 Accepted 31ms 1.027 MiB
#7 Accepted 30ms 1.277 MiB
#8 Accepted 31ms 1.277 MiB
#9 Accepted 12ms 788.0 KiB

Code

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int t;
    cin >> t;

    while (t--) {
        int N, K, D;
        cin >> N >> K >> D;

        vector<int> A(N);
        for (int i = 0; i < N; ++i)
            cin >> A[i];

        int bestIndex = -1;
        int maxProduct = -1;

        int windowSum = 0;
        int zeroCount = 0;

        for (int i = 0; i < K; ++i) {
            windowSum += A[i];
            if (A[i] == 0) zeroCount++;
        }

       
        if (windowSum % D == 0) {
            int currentProduct = (zeroCount == 0) ? 1 : 0;
            if (currentProduct > maxProduct) {
                maxProduct = currentProduct;
                bestIndex = 0;
            }
        }

       
        for (int i = K; i < N; ++i) {
            // Remove A[i - K], add A[i]
            if (A[i - K] == 0) zeroCount--;
            windowSum -= A[i - K];

            if (A[i] == 0) zeroCount++;
            windowSum += A[i];

            if (windowSum % D == 0) {
                int currentProduct = (zeroCount == 0) ? 1 : 0;
                if (currentProduct > maxProduct) {
                    maxProduct = currentProduct;
                    bestIndex = i - K + 1;
                }
                
                else if (currentProduct == maxProduct && bestIndex > i - K + 1) {
                    bestIndex = i - K + 1;
                }
            }
        }

        if (bestIndex == -1)
            cout << -1 << endl;
        else
            cout << (bestIndex + 1) << endl; // Convert to 1-based index
    }

    return 0;
}

Information

Submit By
Type
Submission
Problem
P1190 Segment Strength
Contest
Brain Booster #9
Language
C++11 (G++ 13.2.0)
Submit At
2025-04-06 17:00:09
Judged At
2025-04-06 17:00:09
Judged By
Score
100
Total Time
32ms
Peak Memory
1.277 MiB