/ SeriousOJ /

Record Detail

Time Exceeded


  
# Status Time Cost Memory Cost
#1 Accepted 2ms 1.023 MiB
#2 Accepted 1ms 1.277 MiB
#3 Accepted 170ms 1.191 MiB
#4 Accepted 128ms 1.277 MiB
#5 Accepted 221ms 1.16 MiB
#6 Accepted 56ms 1.145 MiB
#7 Accepted 60ms 1.148 MiB
#8 Accepted 88ms 1.145 MiB
#9 Accepted 104ms 1.152 MiB
#10 Accepted 32ms 1.152 MiB
#11 Accepted 15ms 1.145 MiB
#12 Accepted 870ms 1.199 MiB
#13 Time Exceeded ≥1085ms ≥1.027 MiB
#14 Time Exceeded ≥1075ms ≥1004.0 KiB
#15 Time Exceeded ≥1081ms ≥1.027 MiB
#16 Time Exceeded ≥1067ms ≥1008.0 KiB
#17 Time Exceeded ≥1058ms ≥1.156 MiB

Code

#include <bits/stdc++.h>

#pragma GCC optimize("Ofast")
using namespace std;

const int mxN = 1e4 + 2;
vector<vector<int>> g(mxN);
vector<vector<int>> who(mxN);
vector<int> Par(mxN), siz(mxN), rnk(mxN);

int Leader(int x) {
  if (Par[x] != x) {
    Par[x] = Leader(Par[x]);
  }
  return Par[x];
}

bool unite(int x, int y) {
  int X = Leader(x);
  int Y = Leader(y);

  if (X != Y) {
    if (rnk[X] > rnk[Y]) {
      Par[Y] = X;
      siz[X] += siz[Y];
    } else if (rnk[X] < rnk[Y]) {
      Par[X] = Y;
      siz[Y] += siz[X];
    } else {
      Par[Y] = X;
      siz[X] += siz[Y];
      ++rnk[X];
    }
    return true;
  }
  return false;
}

void solve(int cs) {
  int n, k;
  cin >> n >> k;
  vector<int> a(n);
  for (int i = 0; i < n; i++) {
    cin >> a[i];
  }
  for (int i = 0; i < n; i++) {
    Par[i] = i, rnk[i] = 0, siz[i] = 1;
  }
  for (int i = 0; i < n; i++) {
    for (int j = i + 1; j < n; j++) {
      if (__gcd(a[i], a[j]) > 1) {
        unite(i, j);
      }
    }
  }

  for (int i = 0; i < n; i++) {
    g[Leader(i)].push_back(i);
  }

  for (int i = 0; i < n; i++) {
    for (auto id : g[i]) {
      who[i].push_back(a[id]);
    }
    sort(who[i].rbegin(), who[i].rend());
  }

  int64_t res = 0;
  for (int i = 0; i <= n - k; i++) {
    int64_t sum = 0;
    for (int j = 0; j < n; j++) {
      int f = 0;
      for (auto id : g[j]) {
        if (id >= i && id < i + k) {
          sum += who[j][f++];
        }
      }
    }
    res = max(res, sum);
  }
  cout << res << "\n";
  for (int i = 0; i < n; i++) {
    g[i].clear(), who[i].clear();
  }
}

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
  int tc = 1;
  cin >> tc;
  for (int cs = 1; cs <= tc; cs++) {
    solve(cs);
  }
  return 0;
}

Information

Submit By
Type
Submission
Problem
P1063 Another Maximum Sum in Subarray
Language
C++20 (G++ 13.2.0)
Submit At
2024-07-15 07:25:37
Judged At
2024-07-15 07:25:37
Judged By
Score
70
Total Time
≥1085ms
Peak Memory
≥1.277 MiB