CQBZOJ - 2916 学生宿舍 动态规划(DP) 重庆一中高2018级竞赛班第十次测试 2016.9.16 Problem 1

【问题描述】
新的学生宿舍开放了，它由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;
}