이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5,INF=2e9;
int n,u,v,w,h[N],tot,f[N][2],g[N][2],ans,F0,F1,a[N],b[N],c[N];
struct edge {int to,val,nxt;}e[N<<1];
void add(int u,int v,int w)
{
e[++tot]={v,w,h[u]};
h[u]=tot;
}
void dfs1(int u,int fa)
{
f[u][1]=-INF;
for(int i=h[u];i;i=e[i].nxt)
{
int v=e[i].to,w=e[i].val;
if(v==fa) continue;
c[v]=w;dfs1(v,u);
a[v]=max(f[v][0],f[v][1]+w);
b[v]=w+f[v][0]-a[v];
f[u][1]=max(f[u][1],b[v]);
f[u][0]+=a[v];
}
f[u][1]+=f[u][0];
}
int F(int a,int b,int w) {return w+a-max(a,b+w);}
void dfs2(int u,int fa)
{
int mx=-INF,sec=-INF;
for(int i=h[u];i;i=e[i].nxt)
{
int v=e[i].to;
if(v==fa) continue;
if(b[v]>mx) sec=mx,mx=b[v];
else sec=max(sec,b[v]);
}
int val=F(g[u][0],g[u][1],c[u]),sum=max(g[u][0],g[u][1]+c[u]);
ans=max(ans,f[u][0]+sum);
for(int i=h[u];i;i=e[i].nxt)
{
int v=e[i].to;
if(v==fa) continue;
g[v][0]=f[u][0]-a[v]+sum,g[v][1]=g[v][0];
if(b[v]==mx) g[v][1]+=max(sec,val);
else g[v][1]+=max(mx,val);
dfs2(v,u);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&u,&v,&w);
add(u,v,w),add(v,u,w);
}
dfs1(1,0);c[1]=-INF;dfs2(1,0);
printf("%d",ans);
return 0;
}
컴파일 시 표준 에러 (stderr) 메시지
beads.cpp: In function 'int main()':
beads.cpp:51:7: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
51 | scanf("%d",&n);
| ~~~~~^~~~~~~~~
beads.cpp:54:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
54 | scanf("%d%d%d",&u,&v,&w);
| ~~~~~^~~~~~~~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |