P2279 [HNOI2003]消防局的設立 題解
阿新 • • 發佈:2021-10-17
題面
solve
今天模擬賽考到一道一模一樣的題目,但我還寫了1h+,真的是越來越菜了
其實這道題一股樹形 \(DP\) 的味道,但是要考慮很多狀態,父親,兒子,兄弟,所以考慮貪心
對於一顆以 \(x\) 為根節點的子樹,我們考慮幾種情況
-
需要父節點和兄弟節點給一點貢獻
-
可以給父節點和兄弟節點一點貢獻
-
自洽,給不了父親和兄弟一點東西
我們實際處理的時候可以不分這三種情況,開 \(should[x]\) 和 \(give[x]\) 表示需要的貢獻和能給出的貢獻
然後分類討論一下,看一下需不需要放就好了
顯然,越往上面放坑定好,所以能不放就不放,只有非不得已的時候再放
code
#include<bits/stdc++.h> using namespace std; const int maxn=1e6+5,maxe=2e6+5,INF=0x3f3f3f3f; int N,ans; int lnk[maxn],son[maxe],nxt[maxe],fa[maxn],cnt,shod[maxn],giv[maxn]; struct IO{ static const int S=1<<21; char buf[S],*p1,*p2;int st[105],Top; ~IO(){clear();} inline void clear(){fwrite(buf,1,Top,stdout);Top=0;} inline void pc(const char c){Top==S&&(clear(),0);buf[Top++]=c;} inline char gc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;} inline IO&operator >> (char&x){while(x=gc(),x==' '||x=='\n'||x=='r');return *this;} template<typename T>inline IO&operator >> (T&x){ x=0;bool f=0;char ch=gc(); while(ch<'0'||ch>'9'){if(ch=='-') f^=1;ch=gc();} while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=gc(); f?x=-x:0;return *this; } inline IO&operator << (const char c){pc(c);return *this;} template<typename T>inline IO&operator << (T x){ if(x<0) pc('-'),x=-x; do{st[++st[0]]=x%10,x/=10;}while(x); while(st[0]) pc('0'+st[st[0]--]);return *this; } }fin,fout; inline void add_e(int x,int y){ son[++cnt]=y;nxt[cnt]=lnk[x];lnk[x]=cnt; } void push(int x){ giv[x]=2;shod[x]=-INF; } inline int max(int x,int y){return x>y?x:y;} void DFS(int x){ shod[x]=-INF;giv[x]=-INF; for(int j=lnk[x];j;j=nxt[j])if(son[j]!=fa[x]){ DFS(son[j]); shod[x]=max(shod[x],shod[son[j]]); giv[x]=max(giv[x],giv[son[j]]); } if(shod[x]==2){ ans++;push(x);return ; } if(shod[x]==1){ if(giv[x]>1){shod[x]=-INF;giv[x]--;return ;} if(giv[x]==1) {giv[x]=-INF;shod[x]++;return ;} if(giv[x]==-INF) {shod[x]++;return ;} } if(shod[x]==-INF){ if(giv[x]==-INF){shod[x]=1;return ;} if(giv[x]==1){giv[x]=-INF;return ;} if(giv[x]==2){giv[x]--;return ;} } return; } int main(){ freopen("A.in","r",stdin); freopen("A.out","w",stdout); fin>>N; for(int i=1;i<N;i++) {fin>>fa[i+1];add_e(i+1,fa[i+1]);add_e(fa[i+1],i+1);} DFS(1);if(shod[1]!=-INF)ans++; fout<<ans<<'\n'; return 0; }