vijos1488 路灯改建计划（重庆一中高2018级信息学竞赛测验9） 解题报告

【问题描述】  
  
　　在华师一的敏行路上，新建了若干漂亮的路灯，这给同学们晚上的出行带来很大的方便。但是，问题随之出现了。


　　一天晚上，OI组的FHH 同学正往校门外走，忽然眼前一片漆黑，于是直接把眼镜都摔掉了，再也找不到。后来FHH 同学从学校管理处了解到昨晚路灯突然熄灭是因为电路不堪重负，导致空气开关跳闸。


　　善于思考的FHH 同学考虑将路灯进行改建，以避免再次出现类似的问题。FHH同学仔细了解每盏路灯的耗电量a[i]与照明度z[i]，已知共有 N 盏电灯，并且每盏电灯都可能有不同的耗电量与照明度，现在的问题是要把这 N 盏电灯分为 M 组，新分出的每组灯的耗电量（即是该组所有打开电灯的耗电量之和）不能超过该组的电灯数目的 T 倍，在满足这样的前提下使得照明度尽可能的大，最后算出 M 组的最大照明度的和。由于每组耗电量的限制，该组中的某些电灯可能不被使用，但是仍然应该算作该组灯的数目。特别注意的是电灯按顺序给出，只能把相邻的几盏灯分在一组。


　　由于计算较为复杂，FHH 同学经过反复的计算仍然不能确定结果，现在就请你为他编写一个程序来解决这个问题。 
 
    
 【输入格式】  
  
　　第一行3个整数，分别表示N、M 和 T。
　　接下来的N行，每行两个整数，第i+1行表示a[i]和z[i]。


 
    
 【输出格式】  
   
　　一个整数，表示最大照明度。
 
    
 【输入样例】   
   
5 2 2
1 1
2 2
3 3
4 4
5 5


 
    
 【输出样例】  
   
10
 
    
 【数据范围】  
   
　　对于70％的数据，保证有：2<=N<=80，1<=M<=35，1<=T，a[i]，z[i]<=35；
　　对于全部的数据，保证有：2<=N<=160，1<=M<=50，1<=T，a[i]，z[i]<=50。

做题思路（70分解法）：根据题意，要将N盏电灯分为M组，并且每组中的电灯可以开，可以不开，这显然是分组背包问题，主要算法自然是动态规划，但需要注意分组背包问题的需要两次动态规划，分别是求组权值，即求每组中的电灯选择一些打开，在满足条件的情况下的最大照明度的和，和求将电灯分组时的最大照明度的和。

对于第一次动态规划，设状态函数f(i,j)表示前i个电灯选择一些，耗电量为j时的最大照明度的和，分析第i个电灯时，它可以开，可以不开，所以状态转移方程为f(i,j)=max{f(i-1,j),f(i-1,j-a[i])+z[i]} ，边界条件是f(0,0)=0。为了优化程序，可以先计算出组权值g[i][j]，表示将第i个电灯到第j个电灯放在一组时满足条件的最大照明度的和，在第二次动态规划时，可以直接引用。对于最简单的计算组权值，就可以直接枚举第x个电灯到第y个电灯，每枚举一次，进行一次动态规划。

对于第二次动态规划，设状态函数f(i,j)表示前i个电灯，分为j组时满足条件的最大照明度的和，分析第j组时，最少要有1个电灯，最多只能有i-j+1个电灯（为了保证前j-1组，至少每组有1个电灯），所以状态转移方程为f(i,j)=max{f(i-x,j-1)+g[i-x+1][i] | 1<=x<=i-j+1} ，边界条件是f(0,0)=0。答案即为f(N,M)。

但这种做法在计算组权值时，本来已经要枚举从第x个路灯到第y个路灯了，在进行动态规划时又有两重循环，时间复杂度为O(N^4*M)。

解题思路（正解）：由于直接枚举计算组权值时有四重循环，会导致超时，所以要考虑是否可以减少一重循环，或者将一重循环移在枚举进行动态规划外面来进行优化。因为第一次动态规划的状态函数里限制了前i个电灯和耗电量为j，所以其实在计算组权值时枚举的到第y个路灯的这一重循环，可以移在枚举进行动态规划的外面来，只需枚举起始的路灯编号x，在进行动态规划后，g[x][y]=f(y,T*(y-x+1))（注：这里的状态函数f(i,j)的i是从x开始枚举的，意义与之前的状态函数有所不同，但状态转移方程依然一样）。这样计算组权值的话，时间复杂度就优化为O(N^3*M)。

#include<cstdio>#include<cstdlib>#include<iostream>#include<cstring>#include<algorithm>using namespace std;const int maxn=165;const int inf=1000000010;int N,M,T;int a[maxn],z[maxn];int g[maxn][maxn]; //g[i][j]表示将第i个电灯到第j个电灯放在一组时满足条件的最大照明度的和，即组权值/*f(i,j)表示从第x个路灯到第i个电灯选择一些，耗电量为j时的最大照明度的和f(i,j)=max{f(i-1,j),f(i-1,j-a[i])+z[i]} 边界：f(0,0)=0 */ int f[maxn][maxn*52]; int ready(int x)  //第一次动态规划{memset(f,0,sizeof(f));for(int i=x;i<=N;i++)for(int j=0;j<=T*(N-x+1);j++){int t1=f[i-1][j],t2=0;if(j-a[i]>=0)  t2=f[i-1][j-a[i]]+z[i];f[i][j]=max(t1,t2);}}/*f(i,j)表示前i个电灯，分为j组时满足条件的最大照明度的和f(i,j)=max{f(i-x,j-1)+g[i-x+1][i] | 1<=x<=i-j+1} 边界：f(0,0)=0 */int d[maxn][55];void solve()  //第二次动态规划{for(int i=0;i<=N;i++)for(int j=0;j<=M;j++)d[i][j]=-inf;d[0][0]=0;for(int i=1;i<=N;i++)for(int j=1;j<=M;j++){int t=-inf;for(int x=1;x<=i-j+1;x++)t=max(d[i-x][j-1]+g[i-x+1][i],t);d[i][j]=t;}printf("%d\n",d[N][M]);}int main(){//freopen("light.in","r",stdin);//freopen("light.out","w",stdout);scanf("%d%d%d",&N,&M,&T);for(int i=1;i<=N;i++)scanf("%d%d",&a[i],&z[i]);for(int i=1;i<=N;i++){ready(i);for(int j=i;j<=N;j++)g[i][j]=f[j][T*(j-i+1)];  //计算组权值}solve();return 0;}

考后反思：对于程序的时间优化还是要加强练习，在考试的时候总是不能准确地找到程序的时间优化方法。