T1写炸了今天,期望70却落了个20...连链上的都没有写对
T3什么什么线段树分治套AC自动机,表示我完全自闭了,幸好考场上没有杠T3
总体比赛还是比较舒服,暴力分给的蛮足的,不像昨天那样
T1:
题目链接:
http://172.16.0.132/senior/#main/show/5913
题目:
里口福因有林下风气,带领全国各地高校掀起了一股AK风,大家都十分痴迷于AK。里口福为了打击大家的自信心,出了一道自以为十分困难的题目。
里口福有一棵树,第i个节点上有点权ai,他的问题就是这棵树中有多少个不同的连通块满足连通块的最大值与最小值之差=k,两个连通块不同当且仅当至少存在一个节点在一个连通块中出现而另一个连通块中没有出现。
痴迷于AK的你马上接下这道题目,在里口福狂妄的笑声中,你切掉这道题的决心更加坚定了,现在就差你的代码了。
题解:
我们枚举一下最小值,这样最大值就是唯一确定的了
以最小值为根跑$dp$,$dp(x,r,k)$表示最小值是$a[r]$,当前在$x$节点,最大值和最小值不超过$k$的连通块的个数
当$a[r]<=a[y]<=a[r]+k$时可以转移
发现这样以两个a值相等的点为根可能方案会算重,所以当$a[r]==a[y]$时我们强制让y<r的时候才可以转移
显然若当前枚举的最小值为a[i],对答案的贡献就是$dp(i,i,k)-dp(i,i,k-1)$
#include<algorithm> #include<cstring> #include<iostream> #include<cstdio> using namespace std; const int N=3333+15; const int mo=19260817; int n,k,tot; int a[N],head[N]; struct E{ int to,nxt; }edge[N<<1]; inline int read() { char ch=getchar(); int s=0,f=1; while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();} while (ch>='0'&&ch<='9') {s=(s<<3)+(s<<1)+ch-'0';ch=getchar();} return s*f; } void link(int u,int v){edge[++tot]=(E){v,head[u]};head[u]=tot;} int dp(int x,int pre,int r,int k) { if (k<0) return 0; int re=1; for (int i=head[x];i;i=edge[i].nxt) { int y=edge[i].to; if (y==pre) continue; if (a[y]>=a[r]&&a[y]<=a[r]+k&&(a[y]!=a[r]||(a[y]==a[r]&&y<r))) { re=1ll*re*(dp(y,x,r,k)+1)%mo; } } return re; } int main() { freopen("lkf.in","r",stdin); freopen("lkf.out","w",stdout); n=read();k=read(); for (int i=1;i<=n;i++) a[i]=read(); for (int i=1,u,v;i<n;i++){ u=read();v=read(); link(u,v);link(v,u); } int ans=0; for (int i=1;i<=n;i++) { ans=(ans+((dp(i,0,i,k)-dp(i,0,i,k-1))%mo+mo)%mo)%mo; } printf("%d\n",ans); return 0; }