/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 540.0 KiB
#2 Accepted 9ms 572.0 KiB
#3 Accepted 15ms 540.0 KiB
#4 Accepted 13ms 540.0 KiB
#5 Accepted 13ms 792.0 KiB
#6 Accepted 14ms 1.336 MiB
#7 Accepted 14ms 2.309 MiB
#8 Accepted 14ms 2.066 MiB
#9 Accepted 5ms 788.0 KiB

Code

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <map>
#include <set>
//#include <queue>
//#include <numeric>
#include <cassert>

using namespace std;

#ifdef LOCAL_DEBUG
#include <local_debug.h>
#define DEBUG(...) DBG2::cprint(#__VA_ARGS__, __LINE__, __VA_ARGS__)
#else
#define DEBUG(...)
#endif

#define SZ(a) int((a).size())
#define REP(i,n) for(int i=0,_n=(n);i<_n;++i)
#define FOR(i,a,b) for(int i=(a),_b=(b);i<=_b;++i)

using llong = long long;
using VI = vector<int>;
using VVI = vector<VI>;
using II = pair<int,int>;

int solve(int N, VI A, int K, int D) {
   int sum = 0;
   for (int i = 0; i < K; i++)
      sum += A[i];

   int res = -1;
   int max_prod = -1;
   
   if ((sum % D) == 0) {
      max_prod = sum == K ? 1 : 0;
      res = 1;
   }

   FOR(i, K, N-1) {
      sum += A[i];
      sum -= A[i-K];

      if ((sum % D) == 0) {
         int prod = sum == K ? 1 : 0;
         if (prod > max_prod) {
            max_prod = prod;
            res = i-K+2;
         }
      }
   }
   return res;
}

int main(int argc, char* argv[]) {
   ios_base::sync_with_stdio(false); 
   cin.tie(nullptr);

   int TC;
   cin >> TC;
   FOR(tc, 1, TC) {
      int N, K, D;
      cin >> N >> K >> D;
      VI A(N);
      REP(i, N)
         cin >> A[i];
      int res = solve(N, A, K, D);
      cout << res << '\n';
   }

   return 0;
}

Information

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