主页 > 开源代码  > 

CF1801D

CF1801D

CF1801D 题目大意: n n n 个顶点, m m m 条边的图。你一开始在起点 1,拥有 P P P 枚硬币,通过每条 ( i , j ) (i,j) (i,j) 边都需要花费一定的硬币 s ( i , j ) s(i,j) s(i,j)。但你在每个城市 i i i 都可以打工赚硬币 w i w_i wi​(可以多次打工)。请问从 1 到 n n n 的最少打工次数是多少次?

会有一个贪心的想法:去赚钱最多的城市打最少的工,然后一次性去到终点。但很快就能否决掉,在去赚硬币最多的城市的过程中可能要打很多次工。所以,只在一个城市打工的念头要彻底消除。而去哪打工?打多少次工最好?这是我们需要平衡的地方。因此,这应该是一个动态规划。

方法 1:

使用动态规划分析该问题,设最少打工次数的方案为: { 1 , a 1 , a 2 , . . . , a k , n } \{1,a_1,a_2,...,a_k,n\} {1,a1​,a2​,...,ak​,n}。 当 a k = i a_k=i ak​=i ( i i i 是 n n n 的邻接点):方案变为 { 1 , a 1 , a 2 , . . . , a k − 1 , i , n } \{1,a_1,a_2,...,a_{k-1},i,n\} {1,a1​,a2​,...,ak−1​,i,n}。 方案的属性:

打工次数。原方案的打工次数 = 子方案 { 1 , a 1 , a 2 , . . . , a k − 1 , i } \{1,a_1,a_2,...,a_{k-1},i\} {1,a1​,a2​,...,ak−1​,i} 的打工次数 + i i i 到 n n n 的打工次数。由于,原方案要求打工次数最少。因此,子方案的打工次数也应该越少越好。硬币数量。原方案的剩余的硬币数量 = 子方案 { 1 , a 1 , a 2 , . . . , a k − 1 , i } \{1,a_1,a_2,...,a_{k-1},i\} {1,a1​,a2​,...,ak−1​,i} 剩余的硬币数量 + i i i 到 n n n 的打工次数 * w i w_i wi​ - s ( i , n ) s(i,n) s(i,n)。由于,并不知道原方案剩余多少硬币、 i i i 到 n n n 的打工次数打了多少次工,导致我们无法知道子方案剩余的硬币数量。目的地。子方案目的地变为了 i i i。

根据上述原因,我们的想法是:第一,需要给原问题多一维 剩余硬币 的状态。第二,在方案分析中,添加对打工次数的分类讨论。但是,由于剩余硬币可能是 0 ~ 1 0 9 10^9 109,不可能作为状态使用。对打工次数的分类讨论也无法实现,最多打多少次工呢?即使知道了,时间复杂度也无法通过题目。

此处就是题目较难的思维点:方案中,打工的地方,越往后打工的单价只会越来越高。因为,如果你后面打工的单价低,那还不如在前面先打好工呢。根据此,我们可得到,每一次向后走,用前面打工钱数最高的单价去打工,并且只需要尽量的能到达下一个地点即可。因此,打工次数无需枚举,是固定的。

i i i 到 n n n 的打工次数为:( s ( i , n ) s(i,n) s(i,n) - 子方案剩余的硬币数量) / max ⁡ { w } \max\{w\} max{w}。为了后续的打工次数少,子方案硬币能剩余的越多越好。

同时,打工次数被确定了后,每一次剩余的硬币数量也能被确定:

令 i i i 到 n n n 的打工次数为 h h h。原方案的剩余的硬币数量 = 子方案 { 1 , a 1 , a 2 , . . . , a k − 1 , i } \{1,a_1,a_2,...,a_{k-1},i\} {1,a1​,a2​,...,ak−1​,i} 剩余的硬币数量 + h h h * max ⁡ { w } \max\{w\} max{w} - s ( i , n ) s(i,n) s(i,n)。

所以,此时打工次数只剩下与 max ⁡ { w } \max\{w\} max{w} 也有关。而由于,一共只有 n n n 个物品。因此,我们可以增加一维状态用于表示经过的最高的可打工的地点。此时,子方案反映的问题是:从 1 到 i i i ,且打工的最高单价为 j j j,打工的最少次数是多少,同时剩余的硬币最多是多少?

又由于,这是在图论背景下的问题,我们只能选择贡献式递推去解决该问题,也就是使用 dijkstra 算法去递推。时间复杂度为 O ( n m log ⁡ n ) O(nm\log n) O(nmlogn),如果把 dijkstra 变成朴素版本就可以到达 O ( n 3 ) O(n^3) O(n3)。

#include<bits/stdc++.h> using namespace std; typedef long long ll; const int N = 810; int n,m,p; struct edge{ int to,cost; }; vector<edge>v[N]; struct node{ int now;ll time,coin;int maxn; bool operator < (const node &p)const{ if(time == p.time) return coin < p.coin; return time > p.time; } }; ll ans; int flag[N][N],w[N]; node dist[N][N]; void dij(){ for(int i = 0; i <= n; i++){ for(int j = 0; j <= n; j++){ flag[i][j] = 0; dist[i][j] = {0,(ll)1e18,0}; } } priority_queue<node>q; q.push({1,0,p,0}); dist[1][0] = {0,0,p,1}; while(q.size()){ node p = q.top(); q.pop(); int now = p.now; ll time = p.time,coin = p.coin; int maxn = p.maxn; // cout <<now << " " <<time <<" " <<coin << " " << maxn << "GGGG" << endl; if(now == n){ ans = min(ans,p.time); return; } if(flag[now][maxn]) continue; flag[now][maxn] = 1; for(int i = 0; i < v[now].size(); i++){ edge j = v[now][i]; int to = j.to;ll cost = j.cost; int newmaxn; if(w[maxn] < w[now]) newmaxn = now; else newmaxn = maxn; ll newtime,newcoin; if(coin >= cost) newtime = 0,newcoin = coin - cost; else{ newtime = (cost - coin + w[newmaxn] - 1) / w[newmaxn] ; newcoin = coin + newtime * w[newmaxn] - cost; } if(dist[to][newmaxn].time > time + newtime){ dist[to][newmaxn].coin = newcoin; dist[to][newmaxn].time = time + newtime; }else if(dist[to][newmaxn].time == time + newtime){ dist[to][newmaxn].coin = max(dist[to][newmaxn].coin,newcoin); } q.push({to,dist[to][newmaxn].time,dist[to][newmaxn].coin,newmaxn}); } } } int main(){ int t; cin >> t; while(t--){ cin >> n >> m >> p; for(int i = 1; i <= n; i++){ cin >> w[i]; v[i].clear(); } for(int i = 1; i <= m; i++){ int a,b,s; cin >> a >> b >> s; v[a].push_back({b,s}); } ans = 1e18; dij(); if(ans > 1e17) cout << -1 << endl; else cout << ans << endl; } return 0; } 方式 2

使用动态规划分析该问题,直接设最少打工次数的方案为: { 1 , a 1 , a 2 , . . . , a k , n } \{1,a_1,a_2,...,a_k,n\} {1,a1​,a2​,...,ak​,n}。 但是,这个方案中每个地方都打过工,忽略了未打工的地点。因为,上一个打工点直接通过最短路可以到达下一个打工点。这样做的好处是,我们不需要一维状态去表示最高打工单价了,每一次遍历到的点,就应该是最高打工单价。 下述代码使用了 floyd 去求解任意两个顶点之间的最短路。因此,时间复杂度为 O ( n 3 + m log ⁡ n ) O(n^3+m\log n) O(n3+mlogn)

#include<bits/stdc++.h> using namespace std; typedef long long ll; const int N = 810; int n,m,p; struct node{ int now;ll time,coin; bool operator < (const node &p)const{ if(time == p.time) return coin < p.coin; return time > p.time; } }; ll e[N][N]; void floyd(){ for(int k = 1; k <= n; k++){ for(int i = 1; i <= n; i++){ for(int j = 1; j <= n; j++){ e[i][j] = min(e[i][j],e[i][k] + e[k][j]); } } } } int flag[N],w[N]; node dist[N]; void dij(){ for(int i = 1; i <= n; i++){ flag[i] = 0; dist[i] = {0,(ll)1e18,0}; } priority_queue<node>q; q.push({1,0,p}); dist[1] = {0,0,p}; while(q.size()){ node p = q.top(); q.pop(); int now = p.now; ll time = p.time,coin = p.coin; //cout <<now << " " <<time <<" " <<coin << "GGGG" << endl; if(flag[now]) continue; flag[now] = 1; for(int i = 1; i <= n; i++){ int to = i;ll cost = e[now][i]; if(cost > 1e17) continue; ll newtime,newcoin; if(coin >= cost) newtime = 0,newcoin = coin - cost; else{ newtime = (cost - coin + w[now] - 1) / w[now]; newcoin = coin + newtime * w[now] - cost; } if(dist[to].time > dist[now].time + newtime){ dist[to].coin = newcoin; dist[to].time = dist[now].time + newtime; }else if(dist[to].time == dist[now].time + newtime){ dist[to].coin = max(dist[to].coin,newcoin); } q.push({to,dist[to].time,dist[to].coin}); } } } int main(){ int t; cin >> t; while(t--){ cin >> n >> m >> p; for(int i = 1; i <= n; i++){ cin >> w[i]; } for(int i = 1; i <= n; i++){ for(int j = 1; j <= n; j++){ e[i][j] = 1e18; } e[i][i] = 0; } for(int i = 1; i <= m; i++){ int a,b,s; cin >> a >> b >> s; e[a][b] = min(e[a][b],(ll)s); } floyd(); dij(); if(dist[n].time > 1e17) cout << -1 << endl; else cout << dist[n].time << endl; } return 0; }
标签:

CF1801D由讯客互联开源代码栏目发布,感谢您对讯客互联的认可,以及对我们原创作品以及文章的青睐,非常欢迎各位朋友分享到个人网站或者朋友圈,但转载请说明文章出处“CF1801D