转载请注明出处:http://tokitsukaze.live/

题目链接:http://codeforces.com/problemset/problem/999/D


题意:给n个数和一个k,把n个数按%k的余数,分成n/k组,使得每组数的个数都相等。我们一次操作:可以任选某个数,把它的值+1。问操作的最小次数和操作后的序列。


题解:
先把数先按%k的余数分好,记录下标,这里为了方便操作,用了queue而不是vector。记余数为i的数的个数为sz[i],如果sz[i]>n/k,开个栈,把多的那些数存到栈中,如果sz[i]$<$n/k,就把栈中的那些数往里填。先从左往右扫一遍,如果栈中还有多的数,就再扫一遍。

因为操作是把一个数的值+1,而且多的数必须移走。所以多的数往右移一定是最优的。

代码:

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
#include <bits/stdc++.h>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#define mem(a,b) memset((a),(b),sizeof(a))
#define MP make_pair
#define pb push_back
#define fi first
#define se second
#define sz(x) (int)x.size()
#define all(x) x.begin(),x.end()
using namespace std;
#define _GLIBCXX_PERMIT_BACKWARD_HASH
#include <ext/hash_map>
using namespace __gnu_cxx;
struct str_hash{size_t operator()(const string& str)const{return __stl_hash_string(str.c_str());}};
typedef long long ll;
typedef unsigned long long ull;
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PDD pair<double,double>
#define VI vector<int>
#define VL vector<ll>
const int INF=0x3f3f3f3f;
const ll LLINF=0x3f3f3f3f3f3f3f3f;
const double PI=acos(-1.0);
const double eps=1e-6;
const int MAX=2e5+10;
const ll mod=998244353;
ll a[MAX];
queue<int> res[MAX];
PLL st[MAX];
int n,k,tot;
ll ans;
void gao()
{
int i,j,sz;
ll tmp;
for(i=0;i<k;i++)
{
while(sz(res[i])>n/k)
{
st[tot++]=MP(res[i].front(),i);
res[i].pop();
}
if(sz(res[i])==n/k) continue;
sz=sz(res[i]);
for(j=0;tot&&j<n/k-sz;j++)
{
res[i].push(st[tot-1].fi);
tmp=i-st[tot-1].se;
if(tmp<0) tmp+=k;
a[st[tot-1].fi]+=tmp;
ans+=tmp;
tot--;
}
}
}
int main()
{
int i,j;
while(~scanf("%d%d",&n,&k))
{
for(i=0;i<k;i++)
{
while(!res[i].empty()) res[i].pop();
}
for(i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
res[a[i]%k].push(i);
}
ans=0;
tot=0;
gao();
gao();
printf("%lld\n",ans);
for(i=1;i<=n;i++) printf("%lld%c",a[i]," \n"[i==n]);
}
return 0;
}
/*
8 4
0 0 0 0 2 2 2 2
8 4
1 1 1 1 3 3 3 3
6 2
0 0 1 1 1 1
*/