1. 程式人生 > >BZOJ5287 HNOI2018毒瘤(虛樹+樹形dp)

BZOJ5287 HNOI2018毒瘤(虛樹+樹形dp)

  顯然的做法是暴力列舉非樹邊所連線兩點的選或不選,大力dp。考場上寫的是最暴力的O(3n-mn),成功比大眾分少10分。容斥或者注意到某些列舉是不必要的就能讓底數變成2。但暴力的極限也就到此為止。

  每次重新dp做了大量重複的事,考慮從減少重複計算方面優化。先跑一遍沒有限制的樹形dp。將非樹邊所連線的點拎出來建一棵虛樹。注意到虛樹中某點對其父親的貢獻係數(也即由該點到其父親的鏈上點的dp值與其關係)是不變的,那麼dp出係數,依舊暴力列舉非樹邊就可以了。一定程度上與noip2018d2t3有相似之處?

  碼起來非常長(雖然似乎並沒有很難寫),可能是我姿勢不對。注意暴力列舉非樹邊時不應標記其是否強制被選,而是標記強制被選的次數,防止回溯時出問題。

#include<iostream> 
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long
#define N 100050
#define P 998244353
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,p[N],f[N][2],id[N][2],dfn[N],size[N],fa[N][19],deep[N],num[N][2][2],tmp[2][2],tot,cnt,t,u,ans=0; bool vis[N],flag[N]; struct data{int to,nxt; }edge[N<<1]; void addedge(int x,int y){t++;edge[t].to=y,edge[t].nxt=p[x],p[x]=t;} int ksm(int a,int k) { int s=1; for (;k;k>>=1,a=1ll*a*a%P) if (k&1) s=1ll*s*a%P; return s; } int inv(int a){return ksm(a,P-2);} void dfs(int k) { vis[k]=1;f[k][0]=f[k][1]=1;dfn[k]=++tot;size[k]=1; for (int i=p[k];i;i=edge[i].nxt) if (edge[i].to!=fa[k][0]) if (vis[edge[i].to]) { flag[k]=flag[edge[i].to]=1;bool tag=1; for (int j=1;j<=cnt;j++) if (id[j][0]==edge[i].to&&id[j][1]==k) {tag=0;break;} if (tag) cnt++,id[cnt][0]=k,id[cnt][1]=edge[i].to; } else { fa[edge[i].to][0]=k; deep[edge[i].to]=deep[k]+1; dfs(edge[i].to); f[k][0]=1ll*f[k][0]*(f[edge[i].to][0]+f[edge[i].to][1])%P, f[k][1]=1ll*f[k][1]*f[edge[i].to][0]%P; size[k]+=size[edge[i].to]; } } int lca(int x,int y) { if (deep[x]<deep[y]) swap(x,y); for (int j=18;~j;j--) if (deep[fa[x][j]]>=deep[y]) x=fa[x][j]; if (x==y) return x; for (int j=18;~j;j--) if (fa[x][j]!=fa[y][j]) x=fa[x][j],y=fa[y][j]; return fa[x][0]; } namespace virtual_tree { int m,p[N],t,g[N][2],point[N],stk[N],top,cho[N]; struct data{int to,nxt;}edge[N<<1]; void addedge(int x,int y){t++;flag[x]=flag[y]=1;edge[t].to=y,edge[t].nxt=p[x],p[x]=t;} bool cmp(const int&a,const int&b){return dfn[a]<dfn[b];} void build() { for (int i=1;i<=n;i++) if (flag[i]) point[++m]=i; sort(point+1,point+m+1,cmp); stk[top=1]=1; for (int i=(point[1]==1)+1;i<=m;i++) { int x=lca(point[i],stk[top]); if (x==stk[top]) stk[++top]=point[i]; else { while (top>1&&deep[stk[top-1]]>=deep[x]) addedge(stk[top-1],stk[top]),top--; if (stk[top]!=x) addedge(x,stk[top--]),stk[++top]=x; stk[++top]=point[i]; } } while (top>1) addedge(stk[top-1],stk[top]),top--; } void copy(int k) { g[k][0]=f[k][0],g[k][1]=f[k][1]; for (int i=p[k];i;i=edge[i].nxt) { copy(edge[i].to); g[k][0]=1ll*g[k][0]*inv((1ll*num[edge[i].to][0][0]*f[edge[i].to][0]+1ll*num[edge[i].to][0][1]*f[edge[i].to][1])%P)%P; g[k][1]=1ll*g[k][1]*inv((1ll*num[edge[i].to][1][0]*f[edge[i].to][0]+1ll*num[edge[i].to][1][1]*f[edge[i].to][1])%P)%P; } } void dp(int k) { if (cho[k]) g[k][0]=0; for (int i=p[k];i;i=edge[i].nxt) { dp(edge[i].to); g[k][0]=1ll*g[k][0]*((1ll*num[edge[i].to][0][0]*g[edge[i].to][0]+1ll*num[edge[i].to][0][1]*g[edge[i].to][1])%P)%P; g[k][1]=1ll*g[k][1]*((1ll*num[edge[i].to][1][0]*g[edge[i].to][0]+1ll*num[edge[i].to][1][1]*g[edge[i].to][1])%P)%P; } } int getans() { copy(1); dp(1); return (g[1][0]+g[1][1])%P; } void dfs(int k,int s) { if (k>cnt) {ans=((ans+s*getans())%P+P)%P;return;} cho[id[k][0]]++,cho[id[k][1]]++;dfs(k+1,-s); cho[id[k][0]]--,cho[id[k][1]]--;dfs(k+1,s); } } int main() { #ifndef ONLINE_JUDGE freopen("bzoj5287.in","r",stdin); freopen("bzoj5287.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(),m=read(); for (int i=1;i<=m;i++) { int x=read(),y=read(); addedge(x,y),addedge(y,x); } fa[1][0]=1;deep[1]=1;dfs(1); for (int j=1;j<19;j++) for (int i=1;i<=n;i++) fa[i][j]=fa[fa[i][j-1]][j-1]; virtual_tree::build(); for (int i=1;i<=n;i++) if (flag[i]) virtual_tree::point[++u]=i; for (int j=1;j<=u;j++) { int i=virtual_tree::point[j]; num[i][0][0]=num[i][1][1]=1; if (i!=1) { int x=fa[i][0],y=i; do { tmp[0][0]=(num[i][0][0]+num[i][1][0])%P,tmp[0][1]=(num[i][0][1]+num[i][1][1])%P, tmp[1][0]=num[i][0][0],tmp[1][1]=num[i][0][1]; num[i][0][0]=tmp[0][0],num[i][0][1]=tmp[0][1],num[i][1][0]=tmp[1][0],num[i][1][1]=tmp[1][1]; if (!flag[x]) for (int k=p[x];k;k=edge[k].nxt) if (edge[k].to!=y&&edge[k].to!=fa[x][0]) { num[i][0][0]=1ll*num[i][0][0]*(f[edge[k].to][1]+f[edge[k].to][0])%P; num[i][0][1]=1ll*num[i][0][1]*(f[edge[k].to][1]+f[edge[k].to][0])%P; num[i][1][0]=1ll*num[i][1][0]*f[edge[k].to][0]%P; num[i][1][1]=1ll*num[i][1][1]*f[edge[k].to][0]%P; } y=x,x=fa[x][0]; }while(!flag[y]); } } virtual_tree::dfs(1,1); cout<<ans; return 0; }