-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathC.cpp
89 lines (87 loc) · 1.92 KB
/
C.cpp
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#include<bits/stdc++.h>
using namespace std;
#define fo(i,n) for(i=0;i<n;i++)
#define fo2(i,start,end) for(i=start;i<=end;i++)
#define rfo(i,n) for(i=n-1;i>=0;i--)
#define lli long long int
#define deb(x) cout<<#x<<"="<<x<<endl
#define deb2(x,y) cout<<#x<<"="<<x<<","<<#y<<"="<<y<<endl
#define pb push_back
#define mp make_pair
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define sortall(x) sort(all(x))
#define tr(it,a) for(auto it=a.begin(); it!=a.end();it++)
#define nl '\n'
#define PI 3.1415926535897932384626
#define mod 1000000007
typedef pair<int, int> pii;
typedef pair<lli, lli> pl;
typedef vector<int> vi;
typedef vector<lli> vl;
typedef vector<pii> vpii;
typedef vector<pl> vpl;
bool comp(pii a,pii b){
if(a.F==b.F){
a.S<b.S;
}
return a.F>b.F;
}
void solve(){
int i,j;
int n;
cin>>n;
vpii a;
int temp,sumx=0,sumy=0;
vector<int> vx,vy;
fo(i,n){
cin>>temp;
a.pb({temp,i+1});
// deb(a[i]);
}
sort(all(a),comp);
for(i=0;i<n;i+=2){
if(i==n-1){
// deb2(sumx,sumy);
if(sumx<=sumy){
vx.pb(a[i].S);
}
else{
vy.pb(a[i].S);
}
break;
}
if(sumx<=sumy){
vx.pb(a[i].S);
vy.pb(a[i+1].S);
sumx+=a[i].F;
sumy+=a[i+1].F;
}
else{
vy.pb(a[i].S);
vx.pb(a[i+1].S);
sumy=a[i].F;
sumx=a[i+1].F;
}
}
cout<<vx.size()<<nl;
fo(i,vx.size()){
cout<<vx[i]<<" ";
}cout<<nl;
cout<<vy.size()<<nl;
fo(i,vy.size()){
cout<<vy[i]<<" ";
}cout<<nl;
}
int main(){
// freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int t=1;
// cin>>t;
while(t--){
solve();
}
return 0;
}