首页 > 代码库 > bzoj 1791 DP
bzoj 1791 DP
首先对于一棵树我们可以tree_dp来解决这个问题,那么对于环上每个点为根的树我们可以求出这个树的一端为根的最长链,并且在tree_dp的过程中更新答案。那么我们对于环,从某个点断开,破环为链,然后再用DP来解决这个问题。
备注:很久之前的一道题,刚转的c++,然后T了,也懒得改了。
/************************************************************** Problem: 1791 User: BLADEVIL Language: C++ Result: Time_Limit_Exceed ****************************************************************/ //By BLADEVIL #include <cstdio> #include <cstring> #define maxn 1000010 #define LL long long using namespace std; LL n,time,l; LL other[maxn<<1],last[maxn],pre[maxn<<1],dfn[maxn],low[maxn],vis[maxn],que[maxn],a[maxn],xx[maxn]; LL ans; LL len[maxn<<1],max1[maxn],max2[maxn],sum[maxn],w[maxn],yy[maxn],maxlen[maxn]; LL save; void getmin(LL &x,LL y) {if (y<x) x=y;} void connect(LL x,LL y,LL z) { pre[++l]=last[x]; last[x]=l; other[l]=y; len[l]=z; //if (y>x) printf("%d %d %lld\n",x,y,z); } void dfs(LL x,LL fa) { dfn[x]=low[x]=++time; for (LL q=last[x];q;q=pre[q]) { if (other[q]==fa) continue; if (!low[other[q]]) { dfs(other[q],x); getmin(low[x],low[other[q]]); } else getmin(low[x],dfn[other[q]]); } if (low[x]!=dfn[x]) save=low[x]; } void dp(LL x) { memset(que,0,sizeof que); LL h=0,t=1ll; que[1]=x; vis[x]=1ll; while (h<t) { LL cur=que[++h]; for (LL q=last[cur];q;q=pre[q]) { if (low[other[q]]==low[x]) continue; if (vis[other[q]]) continue; vis[other[q]]=vis[cur]+1ll; que[++t]=other[q]; } } //for (LL i=1;i<=t;i++) printf("%d ",que[i]); printf("\n"); for (LL i=t;i;i--) for (LL q=last[que[i]];q;q=pre[q]) { if (low[other[q]]==low[x]) continue; if (vis[other[q]]!=vis[que[i]]+1) continue; //printf(" %d %d\n",que[i],other[q]); if (max1[other[q]]+len[q]>max1[que[i]]) max2[que[i]]=max1[que[i]],max1[que[i]]=max1[other[q]]+len[q]; else if (max1[other[q]]+len[q]>max2[que[i]]) max2[que[i]]=max1[other[q]]+len[q]; if (max1[other[q]]+max2[other[q]]>maxlen[que[i]]) maxlen[que[i]]=max1[other[q]]+max2[other[q]]; if (max1[que[i]]+max2[que[i]]>maxlen[que[i]]) maxlen[que[i]]=max1[que[i]]+max2[que[i]]; if (maxlen[que[i]]<maxlen[other[q]]) maxlen[que[i]]=maxlen[other[q]]; } //for (LL i=1;i<=t;i++) printf("%d %lld %lld %lld\n",que[i],max1[que[i]],max2[que[i]],maxlen[que[i]]); } void solve(LL x) { LL now=0ll; dfs(x,-1); //printf("%d",save); if (save) for (LL i=1;i<=n;i++) if (low[i]==save) x=i; save=0; /*if (xx[xx[x]]==x) { ans+=(yy[x]>yy[xx[x]])?yy[x]:yy[xx[x]]; return; }*/ LL cur; for (LL i=1;i<=n;i++) if (low[i]==low[x]) dp(i),cur=i; //printf(" %lld\n",max1[x]); LL t=1; a[t]=cur; while (1) { for (LL q=last[a[t]];q;q=pre[q]) { now=(maxlen[a[t]]>now)?maxlen[a[t]]:now; if (low[other[q]]!=low[x]) continue; if (other[q]==a[t-1]) continue; a[++t]=other[q]; sum[t]=len[q]; break; } if (a[t]==cur) break; } //printf(" %d ",x); //for (LL i=1;i<=t;i++) printf(" %lld %d %d\n",max1[a[i]],a[i],sum[i]); t--; for (LL i=2;i<=t;i++) a[i+t]=a[i],sum[i+t]=sum[i]; t*=2; //for (LL i=1;i<=t;i++) printf(" %lld %d %d\n",max1[a[i]],a[i],sum[i]); for (LL i=1;i<=t;i++) sum[i]+=sum[i-1]; LL len=t>>1ll; memset(que,0,sizeof que); LL l=1,r=1; que[1]=1; for (LL i=2;i<=t;i++) { if (i-que[l]+1>len) l++; w[i]=max1[a[que[l]]]+max1[a[i]]+sum[i]-sum[que[l]]; //printf("w[i]=%d",w[i]); printf(" %d %d\n",i,que[l]); while (l<=r&&(max1[a[i]]-sum[i]>max1[a[que[r]]]-sum[que[r]])) r--; que[++r]=i; //for (LL i=l;i<=r;i++) printf("|%d ",que[i]); printf("\n"); } for (LL i=1;i<=t;i++) now=(w[i]>now)?w[i]:now; //printf(" %lld ",ans); ans+=now; } int main() { scanf("%d",&n); for (LL i=1;i<=n;i++) scanf("%d%lld",&xx[i],&yy[i]); for (LL i=1;i<=n;i++) { if (xx[i]==i) continue; if (xx[xx[i]]==i&&xx[i]>i) yy[i]=(yy[xx[i]]>yy[i])?yy[xx[i]]:yy[i],yy[xx[i]]=-1ll; } for (LL i=1;i<=n;i++) if (yy[i]!=-1) connect(i,xx[i],yy[i]),connect(xx[i],i,yy[i]); for (LL i=1;i<=n;i++) if (!low[i]) solve(i); //for (LL i=1;i<=n;i++) printf(" %d %d %d\n",i,low[i],dfn[i]); printf("%lld\n",ans); return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。