/ SeriousOJ /

Record Detail

Wrong Answer


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 532.0 KiB
#2 Wrong Answer 10ms 568.0 KiB
#3 Wrong Answer 15ms 532.0 KiB

Code

// Author: Shohidur Rahman
#include <bits/stdc++.h>
#define ll long long
#define fast ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;
 
int main() {
    fast;
    int t;
    cin >> t;
    while(t--){
    	int n,k,d;
    	cin >> n >> k >> d;
    	int arr[n];
    	int pre[n], zero[n];
    	int zero_count = 0, sum = 0;
    	for (int i = 0; i < n; ++i)
    	{
    		cin >> arr[i];
    		if(arr[i] == 0) zero_count++;
    		zero[i] = zero_count;
    		sum += arr[i];
    		pre[i] = sum;
    	}
    	int maxi = 0, maxi_count = 0;
    	bool ans = false;
    	for (int i = 0; i < n-k+1; ++i)
    	{
    		if(maxi == 1) break;
    		if(i == 0) {
    			if((pre[i+k-1] - pre[i]) % d == 0){
    			if(zero[i+k-1] - zero[i]) {
    				maxi = 0;
    				maxi_count = i+1;
    				ans = true;
    			}
    			else{
    				maxi = 1;
    				maxi_count = i+1;
    				ans = true;
    			}
    		}
    		}
    		else if((pre[i+k-1] - pre[i-1]) % d == 0){
    			if(zero[i+k-1] - zero[i-1]) {
    				maxi = 0;
    				maxi_count = i+1;
    				ans = true;
    			}
    			else{
    				maxi = 1;
    				maxi_count = i+1;
    				ans = true;
    			}
    		}
    	}
    	if(ans) cout << maxi_count << endl;
    	else cout << -1;
    	
    }
    return 0;
}

Information

Submit By
Type
Submission
Problem
P1190 Segment Strength
Language
C++17 (G++ 13.2.0)
Submit At
2025-04-06 18:13:58
Judged At
2025-04-06 18:13:58
Judged By
Score
0
Total Time
15ms
Peak Memory
568.0 KiB