QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#671475#6695. MatchingSchoolbagAC ✓58ms6288kbC++202.3kb2024-10-24 12:45:232024-10-24 12:45:24

Judging History

你现在查看的是最新测评结果

  • [2024-10-24 12:45:24]
  • 评测
  • 测评结果:AC
  • 用时:58ms
  • 内存:6288kb
  • [2024-10-24 12:45:23]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef map<int, int> mii;
typedef unordered_map<int, int> umii;
typedef pair<int, int> pii;
typedef pair<long long, long long> pll;
typedef vector<int> vi;
typedef vector<vector<int>> vc;
typedef vector<long long> vl;
typedef vector<vector<long long>> vll;
typedef __int128_t int128;
#define rep(x, y, z) for(int x = (y); x <= (z); x++)
#define per(x, y ,z) for(int x = (y); x >= (z); x--)
#define yes cout << "Yes\n"
#define no cout << "No\n"

//#define int long long
//const int N = 1e5 + 5;
//const int mod = 1e9 + 7;
const int inf = -1e9 - 5;

struct dsu{
    vi fa, siz;
    void init(int n){ // 初始化
        fa.resize(n + 1);
        iota(fa.begin(), fa.end(), 0);
        siz.resize(n + 1, 1);
    }
    int find(int x){
        if(x == fa[x]) return x;
        return fa[x] = find(fa[x]);
    }
    void merge(int x, int y){
        x = find(x), y = find(y);
        if(x == y) return;
        if(siz[x] > siz[y]) swap(x, y);
        fa[x] = y, siz[y] += siz[x];
        return;
    }
}d;
void solve(){
    int n; cin >> n;
    d.init(n);
    vector<pii> p(n + 1, {-1, -1});
    rep(i, 1, n){
        cin >> p[i].first;
        p[i].second = p[i].first - i;
    }
    sort(p.begin() + 1, p.end(), [&](pii x, pii y){
        return x.second < y.second;
    });
    rep(i, 1, n){
        int r = i;
        while(p[i].second == p[r].second and r <= n) r++;
        r--;
        rep(j, i, r){
            d.merge(i, j);
        }
        i = r;
    }
    map<int, vi> mp;
    rep(i, 1, n){
        int fa = d.find(i);
        mp[fa].push_back(p[i].first);
    }
    ll ans = 0;
    for(auto [x, y] : mp){
        sort(y.begin(), y.end(), [&](int x, int y){
            return x > y;
        });
        int leny = y.size();
        for(int i = 0; i < leny; i += 2){
            if(i + 1 < leny){
                if(y[i] + y[i+1] > 0){
                    ans += y[i] + y[i+1];
                }
                else break;
            }
        }
    }
    cout << ans << '\n';
}

signed main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int __ = 1; cin >> __;
    while(__--){
        solve();
    }
    return 0;
}

这程序好像有点Bug,我给组数据试试?

详细

Test #1:

score: 100
Accepted
time: 0ms
memory: 3536kb

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: 58ms
memory: 6288kb

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