再练动态规划——(5)Jury Compromise

Jury Compromise

首先看到数据范围就很容易想到动态规划

不过虽然 $M<=20$ ，这道题也显然不容易装压，总共人数 $N<=200$ 。

一开始我想的是$f[i][j]$ 表示到第 $i$ 个人，分差是 $j$ ，可是我太菜写不出状态转移方程，但是杰哥曾经告诉我们：做不出就加一维。
所以就能想到用 $f[i][j][k]$ 表示前 $i$ 个人中选出来 $j$ 个，且分差是 $k$。然后就能推出来方程：

$\bullet$ 如果不选第 $i$ 个人
$$f[i][j][k]=f[i-1][j][k];$$
$\bullet$ 如果选择第 $i$ 个人
$$f[i][j][k]=max(f[i][j][k],f[i-1][j-1][k-a[i]+b[i]]+a[i]+b[i]);$$

由于是取最大值，所以可以先给每一种状态(除了初始状态)都先设个初始值是比较小的负数。

这里的 $a[i]$ 和 $b[i]$ 分别表示第 $i$ 辩方和控方给第 $i$ 个人打的分数
先想到这个式子，然后我们考虑特殊情况，比如如果 $k-a[i]+b[i]$ 如果小于 $0$ 的话会导致数组越界

这时候其实可以引入个变量 $c$ 等于 $k-a[i]+b[i]$ 然后判断下 $c$ 的范围，如果不合法就 continue; 掉。

那么什么情况 $c$ 是合法的呢？
这时候我们就要注意到为什么只选最多 $20$ 个人，每个人最多打分 $20$ 分。
首先每个人是不能打负分的，然后如果差值最大的话是 $20\ast 20=400$ 分，可能是辩方比控方多 $400$ 或者是控方比辩方多 $400$ 。上下都是 $400$，合起来 $800$ 。如果以 $k=400$ 表示给控方和辩方打分一样，那么第三维开个 $800$ ，数组大小是 $200\ast 20\ast 800=3,200,000$ ，除以两次 $1024$ 再乘个 $4$ 算出来空间是十几Mb，每一维再多加个10防越界都够。

引入偏移量 $400$ 就能解决刚刚那个问题，状态转移的时候写个if(c<0||c>800)continue;就彳亍。
其实还有一种状态是到了第 $i$ 个人，选了 $0$ 个，这首先第 $i$ 个人是不选的，用 不选 的那个方程转移一下状态，然后再判断 $j$ 是否为 $0$，若为 $0$ 就continue;掉，不触发 选 的那个方程。
而初始状态的话就设个f[0][0][base]=0;，其他全是负数。

但是为什么不是加个绝对值嘞？
如果我们只是加个绝对值，就可能会出现冲突，比如到第 $i$ 个人时选了 $j$ 个时辩方比控方多 $1$ 分，但我选择另外 $j$ 个的时候控方比辩方多了 $1$ 分，但是方案二的总分低于方案一，这时方案二就会被覆盖掉。

对于差值最小，我们不管是控方分比辩方分高，还是辩方分比控方分高，先设个差值v=0 ，然后每次看$f[n][m][py+v]$ 和 $f[n][m][py-v]$ 是否大于 $0$ 。然后v++;，如果找到合法的就可以记录下来然后continue;掉。

其实好多题都难在输出方案上，那么这道题该怎么做到呢？
我先设 $f[n][m][v]$ 为差值最小的方案，然后回过头来看我们的状态转移方程，如果不选的话是有式子$$f[i][j][k]=f[i-1][j][k];$$
所以可以反过来推，if(f[i][j][k]==f[i-1][j][k])则表示第 $i$ 个人没有被选中，然后我们手写栈或者队列啥的东西记录下来这个 $i$ 就可以了。

时间复杂度大概是 $O(800nmT)$ 左右的样子。
由于这个题多测，别忘了每次清零。
附上代码：

#include<bits/stdc++.h>
using namespace std;
int n,m,v,b[210],a[210],f[210][30][810],py=400,fangan[30],tot;
int main()
{freopen("Jury_Compromise.in","r",stdin);freopen("Jury_Compromise.out","w",stdout);
    int T=1;
    while(1)
	{
		tot=0;
    	scanf("%d%d",&n,&m);
		if(n==0&&m==0)return 0; 
        for(int i=1;i<=n;i++)scanf("%d%d",&a[i],&b[i]);
        memset(f,-0x3f,sizeof f);
        f[0][0][py]=0;
        for(int i=1;i<=n;i++)for(int j=0;j<=m;j++)for(int k=0;k<=800;k++)
		{
            f[i][j][k]=f[i-1][j][k];
            int c=k-a[i]+b[i];
        	if(c<0||c>800||j==0)continue;
            f[i][j][k]=max(f[i][j][k],f[i-1][j-1][c]+a[i]+b[i]);
        }
        v=0;
        while(f[n][m][py+v]<0&&f[n][m][py-v]<0)v++;
        if(f[n][m][py-v]>f[n][m][py+v])v=py-v;
        else v=py+v;
        int i=n,j=m,ans1=0,ans2=0;
        while(j)
		{
            if(f[i][j][v]==f[i-1][j][v])i--;
            else{fangan[++tot]=i;v=v-(a[i]-b[i]);i--;j--;}
        }
        for(int i=1;i<=tot;i++){ans2+=b[fangan[i]];ans1+=a[fangan[i]];}
        printf("Jury #%d\n Best jury has value %d for prosecution and value %d for defence:\n",T++,ans1,ans2);
        for(int i=1;i<=tot;++i){printf(" %d",fangan[i]);}
        puts("\n");
    }
    return 0;
}