1. 程式人生 > >BZOJ4985 評分(二分答案+樹形dp)

BZOJ4985 評分(二分答案+樹形dp)

  首先二分答案簡化一下問題,現在只有0和1了,要求最後剩下的是1。再簡化一下考慮沒有已固定的位置怎麼做。考慮每個位置由其合併到的位置連邊,顯然這樣形成了一棵三叉樹。設f[i]為使得某位置為1其子樹至少要放多少個1即可,轉移顯然。加上已固定位置也類似,修改dp初值即可。

#include<iostream> 
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long #define N 100010 char getc(){char c=getchar();while ((c<'A'||c>'Z')&&(c<'a'||c>'z')&&(c<'0'||c>'9')) c=getchar();return c;} int gcd(int n,int m){return m==0?n:gcd(m,n%m);} int read() { int x=0,f=1;char c=getchar(); while (c<'0'||c>'9') {if
(c=='-') f=-1;c=getchar();} while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x*f; } int n,m,cnt,p[N<<1][3],a[N],b[N],c[N],f[N<<1],ans; void dfs(int k) { if (k<=n) {f[k]=c[k]>=0?(c[k]?0:N):1;return;} dfs(p[k][0]),dfs(p[k][1]),dfs(p[k][2
]); f[k]=min(min(f[p[k][0]]+f[p[k][1]],f[p[k][0]]+f[p[k][2]]),f[p[k][1]]+f[p[k][2]]); } bool check(int k) { memset(f,42,sizeof(f)); memset(c,255,sizeof(c)); for (int i=1;i<=m;i++) c[b[i]]=a[i]>=k?1:0; int tot=0; for (int i=m+1;i<=n;i++) tot+=a[i]>=k; dfs(cnt); return f[cnt]<=tot; } int main() { #ifndef ONLINE_JUDGE freopen("bzoj4985.in","r",stdin); freopen("bzoj4985.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(),m=read(); int l=1,r=0; for (int i=1;i<=m;i++) r=max(r,a[i]=read()),b[i]=read(); for (int i=m+1;i<=n;i++) r=max(r,a[i]=read()); cnt=n; for (int i=1;i<=(n-1>>1);i++) cnt++,p[cnt][0]=i*3-2,p[cnt][1]=i*3-1,p[cnt][2]=i*3; while (l<=r) { int mid=l+r>>1; if (check(mid)) ans=mid,l=mid+1; else r=mid-1; } cout<<ans; return 0; }