-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path1114B.cpp
More file actions
34 lines (33 loc) · 744 Bytes
/
1114B.cpp
File metadata and controls
34 lines (33 loc) · 744 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include<bits/stdc++.h>
#define intl long long
using namespace std;
int main(){
intl n,m,k;
cin>>n>>m>>k;
intl a[n];
intl b[n];
for(int i=0;i<n;i++){cin>>a[i];b[i]=a[i];}
sort(b,b+n,greater<intl>());
unordered_map<intl,int>um;
intl su=0;
for(int i=0;i<m*k;i++){
um[b[i]]++;
su+=b[i];
}
vector<int>v;
int n1=0;
for(int i=0;i<n;i++){
if(um[a[i]]){
um[a[i]]--;
n1++;
// cout<<i<<" "<<a[i]<<"\n";
if(n1==m){
n1=0;
v.push_back(i+1);
}
}
}
cout<<su<<"\n";
for(int i=0;i<v.size()-1;i++)cout<<v[i]<<" ";cout<<"\n";
}
///int a[n];