【问题描述】
新的学生宿舍开放了,它由M栋建筑物构成,标号为1到M。开始时,学生宿舍都是空的,很快有N个学生搬进去了。刚好每天搬进去一个。
每次有新同学搬进宿舍,那栋建筑将会举行一个大型的party。party的噪声和该建筑物里的学生的数量相等。宿舍管理员不喜欢噪声,所以他们会不定期的清空某栋建筑物。清空的方法就是把该栋建筑物的学生全部赶到另外的学生宿舍(这M栋宿舍以外的地方)。但是管理员最多只能清空K次。求这n天噪音之和的最小值。
【输入格式】
第一行三个整数n,m,k。
接下来有n行,每行一个整数bi(1<=bi<=m),表示第i天有一个学生搬进了bi栋宿舍。
【输出格式】
一个整数,表示这n天噪音之和的最小值。
【输入样例】
5 1 2
1
1
1
1
1
【输出样例】
7
【样例解释】
样例1解释:在第一天和第三天之后执行清空操作。这样每天的噪声为1,1,2,1,2,所以噪声之和为7.
【数据范围】
1<=n<=1000000,1<=m<=100,1<=k<=500
【来源】
http://www.cqoi.net:2012/problem.php?id=2916
#include<cstdio>
#include<cstring>
#include<iostream>
#define inf 100000000000000ll
using namespace std;
int n,m,k,num[105];
long long f[105][505],cost[105][505];
int read()
{
int x=0,ok=0;
char ch;
ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
while((ch>='0'&&ch<='9')||ch=='-')
{
if(ch=='-') ok=1;
else x=x*10+ch-'0';
ch=getchar();
}
return ok==1?-x:x;
}
void calc()
{
for(int i=1;i<=m;i++)
{
for(int j=0;j<=k;j++)
{
int t=num[i]/(j+1);
int tt=num[i]%(j+1);
cost[i][j]=((1+t)*t/2)*(j+1)+tt*(t+1);
}
}
}
void solve()
{
calc();
memset(f,10,sizeof(f));
for(int j=0;j<=k;j++)
{
f[0][j]=0;
}
for(int i=1;i<=m;i++)
{
for(int j=0;j<=k;j++)
{
long long t=inf;
for(int z=0;z<=j;z++)
{
t=min(t,f[i-1][j-z]+cost[i][z]);
}
f[i][j]=t;
}
}
cout<<f[m][k]<<endl;
}
int main()
{
//freopen("in.txt","r",stdin);
memset(num,0,sizeof(num));
scanf("%d%d%d",&n,&m,&k);
int x;
for(int i=1;i<=n;i++)
{
x=read();
num[x]++;
}
solve();
return 0;
}
思路:
数据较大,用read()更省时间。
本题有正误两种思路:
第一种:二分答案法+贪心(错误的解法):
二分猜答案尝试每个宿舍人数到了mid人就需要清空一次,看最后需要清空的次数与k进行比较,再更改mid的值。但是错误之处在于并不是每个宿舍到了相同人数再清是最优的解,可简单举例即可得证。
错误代码如下:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int n,m,k,b[1000000],num[105];
int read()
{
int x=0,ok=0;
char ch;
ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
while((ch>='0'&&ch<='9')||ch=='-')
{
if(ch=='-') ok=1;
else x=x*10+ch-'0';
ch=getchar();
}
return ok==1?-x:x;
}
bool check(int mid)
{
memset(num,0,sizeof(num));
int cnt=0;
for(int i=1;i<=n;i++)
{
num[b[i]]++;
if(num[b[i]]>=mid)
{
cnt++;
num[b[i]]=0;
}
}
for(int i=1;i<=m;i++)
{
if(num[i]==0) cnt--;
}
if(cnt>k) return 0;
return 1;
}
int main()
{
//freopen("in.txt","r",stdin);
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)
{
b[i]=read();
}
int left=1,right=n,ans;
while(left<=right)
{
int mid=(left+right)/2;
if(check(mid))
{
right=mid-1;
ans=mid;
}
else
{
left=mid+1;
}
}
long long ans1=0;
int now[105]={0};
for(int i=1;i<=n;i++)
{
now[b[i]]++;
ans1+=now[b[i]];
if(now[b[i]]>=ans)
{
now[b[i]]=0;
}
}
cout<<ans1<<endl;
return 0;
}
第二种:动态规划(正解):详见代码
#include<cstdio>
#include<cstring>
#include<iostream>
#define inf 100000000000000ll
using namespace std;
int n,m,k,num[105];
long long f[105][505],cost[105][505];
int read()
{
int x=0,ok=0;
char ch;
ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
while((ch>='0'&&ch<='9')||ch=='-')
{
if(ch=='-') ok=1;
else x=x*10+ch-'0';
ch=getchar();
}
return ok==1?-x:x;
}
void calc()
{
for(int i=1;i<=m;i++)
{
for(int j=0;j<=k;j++)
{
int t=num[i]/(j+1);
int tt=num[i]%(j+1);
cost[i][j]=((1+t)*t/2)*(j+1)+tt*(t+1);
}
}
}
void solve()
{
calc();
memset(f,10,sizeof(f));
for(int j=0;j<=k;j++)
{
f[0][j]=0;
}
for(int i=1;i<=m;i++)
{
for(int j=0;j<=k;j++)
{
long long t=inf;
for(int z=0;z<=j;z++)
{
t=min(t,f[i-1][j-z]+cost[i][z]);
}
f[i][j]=t;
}
}
cout<<f[m][k]<<endl;
}
int main()
{
//freopen("in.txt","r",stdin);
memset(num,0,sizeof(num));
scanf("%d%d%d",&n,&m,&k);
int x;
for(int i=1;i<=n;i++)
{
x=read();
num[x]++;
}
solve();
return 0;
}