codeforces #538div2

B

题目链接
小红是一个程序媛，她和一般程序员不同的地方是，小红有艺术爱好，她喜欢种花。她种了一些盆栽，每盆盆栽的花瓣数量不同，因为小红是一个爱思考的程序员，她的头发比较少，所以她喜欢花瓣多的花。

在一堆盆栽里面，花瓣数最多的前m盆盆栽被定义为漂亮的花。例如：当m=2时，有4盆花花瓣数为：10 20 5 15，那么漂亮的花是第2和4盆

她现在要把这n盆盆栽搬到k个房间，请问小红要怎么搬这些花，才能使得漂亮的花的花瓣数总和最大？搬花的时候花的顺序不能改变。

当有多种情况符合时，输出任意一种

Input
第一行输入n,m,k (2<=n<=2×10^5),(1<=m,2<=k ， m×k<=n)
第二行输入n个整数表示花的花瓣数a1,a2,……(-10^9<=ai<=109)

Output
第一行输出花瓣数总和
第二行输出每个房间的最后一盆花号，最后一个房间不用输出。

Examples
Input
9 2 3
5 2 5 2 4 1 1 3 2
Output
21
3 5
Input
6 1 4
4 1 3 2 2 3
Output
12
1 3 5
Input
2 1 2
-1000000000 1000000000
Output
0
1
Note
对于第一组样例：
最大漂亮数是21
放入三个房间分别是{5，2，5}，{2，4}，{1，1，3，2}
21=5+5+2+4+3+2

题意

把n个数分成k组，每个组取前m大的数，求它们的总和，打印前k-1个组的最后一个数

思路

排序标记前m*k大的数，之后每m个数输出一个下标

题目还是不难，debug了好久，就是想把自己写的版本debug通过，服了自己，我离AC几公里 [哭笑]，还好最后AC了，独立完成的噢，不然真的要陷入自我怀疑了

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
 
bool mark[N];
struct node
{
    int  zhi;//值
    int w;//下标
 
}a[N];
bool cmp(node &a1,node &a2)//从主函数里引用，就不用额外定义了
{
    if(a1.zhi!=a2.zhi)
       return a1.zhi>a2.zhi;
    return a1.w>a2.w;
}
 
int main()
{
    int n,m,k;
     long long sum=0;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
     {
         cin>>a[i].zhi;
         a[i].w=i;
     }
 
     sort(a+1,a+1+n,cmp);
     for(int i=1;i<=m*k;i++)
     {
         sum+=a[i].zhi;
         int p=a[i].w;
        mark[p]=true;//把前m*k大的数标为true,等下来筛选
     }
 
     cout<<sum<<endl;
     int g=0,ki=0;
     for(int i=1;i<=n&&g<k-1;i++)
        if(mark[i])
     {
         ki++;
 
         if(ki==m)
         {
             cout<<i<<" ";
             ki=0;
             g++;
         }
 
     }
 
 
     cout<<endl;
 
    return 0;
}

另一种写法

#include<bits/stdc++.h>
using namespace std;
int n,m,k,a[200005],id[200005];
long long sum;
bool cmp(int x,int y) {
	return a[x]>a[y];
}
int main() {
	cin>>n>>m>>k;
	int i;
	for(i=1; i<=n; i++) {
		cin>>a[i];
		id[i]=i;
	}
	sort(id+1,id+n+1,cmp);//按照a[i]的大小排序id[i]
	for(i=1; i<=m*k; i++)sum+=a[id[i]];
	cout<<sum<<'\n';
	sort(id+1,id+m*k+1);
	for(i=m; i<m*k; i+=m)cout<<id[i]<<" ";
	return 0;
}