P6883 [COCI2016-2017#3] Kroničan

题目描述

        Mislav 有 𝑁个玻璃杯，从 1∼𝑁 编号，每个玻璃杯中都有一定的水。你需要通过倒水（将某个杯子中的水倒入另一个杯子），使这些杯子中只有 𝐾 个有水。

        已知将第 𝑖号玻璃杯中的水倒入第 𝑗号，需要消耗 𝐶𝑖,𝑗的代价。Mislav 想知道，经过倒水后满足只有 𝐾个（或更少）玻璃杯中有水时，消耗的代价总和的最小值。

        样例

                 对于 40% 的数据，满足 𝑁≤10

                对于 100% 的数据，满足 1≤𝐾≤𝑁≤20,𝐶𝑖,𝑗≤1e5

分析

        看完题目，不难发现，本质上就是一个不断合并的过程。

        看到数据范围 1≤𝐾≤𝑁≤20  ,很容易想到状压DP

        接下来来考虑怎么转移：我们先钦定，一个数二进制的某位为1表示这个杯子的水已经倒出（即该杯子为空），那么显然把水倒入一个没水的杯子一定更劣，所以每次转移只能将一个为0的位置变为1其他位置不变，接着就是枚举从哪种状态转移，转移哪一位，转移至哪一位。还有就是注意到水到出去与倒到哪里没有直接关系，所以

        具体分析见代码

Code

        这里提供2份代码，第一份较好理解，但对空间需求略大，所以用了滚动数组，

第二份是在看完题解后略加修改的（

#include<bits/stdc++.h>
using namespace std;
int n,m,f[2][1050005],a[25][25],ans=1e9;
//f能压的原因是转移的时候，只会用到i-1的状态
//f[i&1][j]表示 出现i个1，状态为j的最小值
vector<int> ve[25];
//ve[i] 表示出现1的个数为i的状态
void dfs(int x,int pos,int num)
{
	if(x==n) { ve[num].push_back(pos); return; }
	dfs(x+1,pos,num);
	dfs(x+1,pos|(1<<x),num+1);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%d",&a[i][j]);
	memset(f,0x3f,sizeof f); f[0][0]=0;
	dfs(0,0,0);//预处理出所有状态
	for(int i=0;i<n;i++)//枚举倒出水的杯数
		for(auto v:ve[i])//枚举当前状态
			for(int j=0;j<n;j++)//枚举第几杯水
				if(((v>>j)&1)==0)//若当前水未倒出
                //注意位运算优先级
				{
					int mi=1e9;
					for(int k=0;k<n;k++)
                        if(((v>>k)&1)==0 && j!=k) //枚举倒到哪里
                            mi=min(mi,a[j+1][k+1]); //去倒处去的的最小代价
					f[(i+1)&1][v|(1<<j)]=min(f[(i+1)&1][v|(1<<j)],f[i&1][v]+mi);
                    //转移
				}
	for(auto v:ve[n-m]) ans=min(ans,f[(n-m)&1][v]);
    //m个未倒出，即倒出了n-m个，枚举找答案
	cout<<ans;
	return 0;
}

#include<bits/stdc++.h>
using namespace std;
int n,m,f[1050005],a[25][25],ans=1e9;
vector<int> ve[25];
void dfs(int x,int pos,int num)
{
	if(x==n) { ve[num].push_back(pos); return; }
	dfs(x+1,pos,num);
	dfs(x+1,pos|(1<<x),num+1);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%d",&a[i][j]);
	memset(f,0x3f,sizeof f); f[0]=0;
	dfs(0,0,0);
	for(int i=0;i<(1<<n);i++)
		for(int j=0;j<n;j++)
			if((i>>j)&1)
				for(int k=0;k<n;k++)
					if(((i>>k)&1)==0) f[i]=min(f[i],f[i^(1<<j)]+a[j+1][k+1]);

	for(auto v:ve[n-m]) ans=min(ans,f[v]);
	cout<<ans;
	return 0;
}