本文主要是介绍货物搜集,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
货物搜集
题解
我们可以先更新出到某个点所需的最大武力值,再二分出总的最小武力值即可。
源码
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<stack>
#include<vector>
#include<queue>
#define MAXN 1000005
using namespace std;
typedef long long LL;
int n,maxx[MAXN],Max,ans;
int to[MAXN],paid[MAXN],nxt[MAXN];
int tot,head[MAXN],a[MAXN];
LL W;
#define gc() getchar()
template<typename _T>
inline void read(_T &x)
{_T f=1;x=0;char s=gc();while(s>'9'||s<'0'){if(s=='-')f=-1;s=gc();}while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=gc();}x*=f;
}
void addEdge(int u,int v,int w)
{to[++tot]=v;paid[tot]=w;nxt[tot]=head[u];head[u]=tot;
}
void dfs(int u,int fa)
{//printf("%d %d\n",u,maxx[u]);for(int i=head[u];i;i=nxt[i]){int v=to[i];if(v==fa) continue;maxx[v]=max(paid[i],maxx[u]);dfs(v,u);}
}
bool tant(int x)
{LL res=0;for(int i=2;i<=n;i++)if(maxx[i]<=x)res+=1ll*a[i];return res>=W;
}//判断当前武力值是否可行
void tann(int l,int r)
{//printf("%d %d\n",l,r);if(l>r) return ;int mid=(l+r)>>1;if(tant(mid)){ans=mid;tann(l,mid-1);}elsetann(mid+1,r);
}//二分
int main()
{ read(n);read(W);for(int i=2;i<=n;i++)read(a[i]);for(int i=1;i<n;i++){int u,v,w;read(u);read(v);read(w);addEdge(u,v,w);addEdge(v,u,w);Max=max(w,Max);}maxx[1]=0;dfs(1,1);//更新整棵树ans=Max;tann(0,Max);//求答案printf("%d",ans);return 0;
}
谢谢!!!
这篇关于货物搜集的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!