UVA 10603Fill 劉汝佳代碼紫書代碼理解

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
using namespace std;
struct Node {
	int v[3], dist;		//三杯水、倒水量
	bool operator < (const Node&rhs)const {
		return dist > rhs.dist;		//用於priority_queue的排序,
									//使得倒水量少的在頂部
	}
};
const int maxn = 200 + 5;
int vis[maxn][maxn], cap[3], ans[maxn];
void update_ans(const Node& u) {//記錄某些水量出現時所需的最低倒水量
	for (int i = 0; i < 3; i++) {
		int d = u.v[i];
		if (ans[d] < 0 || u.dist < ans[d])ans[d] = u.dist;	
		//如果某一水量沒記錄,或又更優量 記錄
	}
}
void solve(int a, int b, int c, int d) {
	cap[0] = a, cap[1] = b, cap[2] = c;	//每個杯子的容量
	memset(vis, 0, sizeof(vis));
	memset(ans, -1, sizeof(ans));
	priority_queue<Node> q;

	Node start;
	start.dist = 0;
	start.v[0] = 0; start.v[1] = 0; start.v[2] = c;	//開始狀態
	q.push(start);
	vis[0][0] = 1;
	while (!q.empty()) {
		Node u = q.top(); q.pop();		//選取倒水量最少
		update_ans(u);
		if (ans[d] >= 0)break;		//找到目標狀態
		for (int i = 0; i < 3; i++) {
			for (int j = 0; j < 3; j++) {		//i水杯往j水杯裏倒水
				if (i != j) {					//不能自己倒給自己
					if (u.v[i] == 0 || u.v[j] == cap[j])continue;	
					//如果i水杯沒水,或者j水杯滿了
					int amount = min(cap[j], u.v[i] + u.v[j]) - u.v[j];		
					//可往j水杯倒的最大水量
					Node u2;
					memcpy(&u2, &u, sizeof(u));		//注意要傳地址
					u2.dist = u.dist + amount;
					u2.v[i] -= amount;
					u2.v[j] += amount;
					if (!vis[u2.v[i]][u2.v[j]]) {	
					//當前狀態是否出現,由於總水量不變,
					//由前兩水杯可以得到第三水杯
						vis[u2.v[i]][u2.v[j]] = 1;
						q.push(u2);
					}
				}
			}
		}
	}
	while (d >= 0) {
		if (ans[d] >= 0) {
			printf("%d %d\n", ans[d], d);	
			//如果記錄到了d狀態的最小倒水量 直接打印
			return;
		}
		d--;		//否則 取最接近的
	}
}
int main() {
	int T, a, b, c, d;
	scanf("%d", &T);
	while (T--) {
		scanf("%d%d%d%d", &a, &b, &c, &d);
		solve(a, b, c, d);
	}
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章