QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#127602 | #6695. Matching | moon_095 | AC ✓ | 139ms | 8012kb | C++17 | 973b | 2023-07-19 20:29:14 | 2023-07-19 20:29:15 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
// #define int long long
long long M = 998244353;
typedef long long LL;
typedef pair<int, int> PII;
int n;
void solve() {
cin >> n;
vector<int> a(n + 1);
for(int i = 1; i <= n; i++) cin >> a[i];
// vector<int>
LL ans = 0;
map<int, int> mp;
vector<LL> b[n + 1];
int idx = 0;
for(int i = 1; i <= n; i++) {
int d = a[i] - i;
if(!mp.count(d)) {
mp[d] = ++idx;
b[idx].push_back(0);
}
b[mp[d]].push_back(a[i]);
}
// for(int i = 1; i <= idx; i++) {
// for(auto c: b[i]) cout << c << ' ';
// cout << endl;
// }
for(int i = 1; i <= idx; i++) {
sort(b[i].begin() + 1, b[i].end());
int r = b[i].size() - 1;
while(r > 1) {
LL rr = b[i][r], ll = b[i][r - 1];
if(ll + rr >= 0) ans += ll + rr;
r -= 2;
}
}
cout << ans << endl;
}
int main() {
int t;
cin >> t;
while(t--) solve();
}
这程序好像有点Bug,我给组数据试试?
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3496kb
input:
3 9 3 -5 5 6 7 -1 9 1 2 3 -5 -4 -3 3 1 10 100
output:
30 0 0
result:
ok 3 number(s): "30 0 0"
Test #2:
score: 0
Accepted
time: 139ms
memory: 8012kb
input:
5504 9 -1 -7 -6 -5 -4 -3 5 -1 0 5 999999995 999999993 999999995 999999995 999999995 5 3 -6 -5 -4 -2 4 -8 2 3 -5 4 -2 -1 0 1 9 -4 -9 3 -1 -1 -5 2 -3 -5 7 -1 -2 1 2 3 4 3 4 -2 5 2 -4 10 2 4 1 -3 -2 4 5 -3 0 -4 6 -1 0 1 2 4 -3 5 -4 -3 -2 -1 0 4 -1 0 1 2 8 1 0 -4 -1 0 -5 -3 -5 2 5 6 8 -4 -3 -2 -1 0 1 2 ...
output:
4 1999999988 0 5 1 1 11 0 9 3 0 3 0 11 6 9 0 1999999989 13 1 11 7 1999999981 40 0 11 0 6 0 0 7 9 1 15 4 3 0 7 23 0 5 1999999997 0 3 5999999976 3 16 5999999943 5999999933 5 0 11 0 10 12 0 6 8 3999999982 12 0 7 0 0 3 3 0 1999999994 3999999972 1 0 16 0 0 0 7 2 0 8 0 5999999964 16 0 1 1999999995 28 0 54...
result:
ok 5504 numbers