这怕是noip的考试范围哦。Achen在半年前就会了的算法我现在都不会。
自己比较智障，连写3道题都是get_root()之后就去solve(z)而不是solve(root)，搞得又T又RE
点分治模板(洛谷)
//Serene
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<set>
using namespace std;
#define ll long long
#define db double
#define For(i,a,b) for(int i=(a);i<=(b);++i)
#define Rep(i,a,b) for(int i=(a);i>=(b);--i)
const int maxn=1e4+7;
int n,m,ans[maxn],p[maxn];
int size[maxn],son[maxn],sumsize,root,vis[maxn],len[maxn];
char cc; ll ff;
template<typename T>void read(T& aa) {
aa=0;cc=getchar();ff=1;
while((cc<'0'||cc>'9')&&cc!='-') cc=getchar();
if(cc=='-') ff=-1,cc=getchar();
while(cc>='0'&&cc<='9') aa=aa*10+cc-'0',cc=getchar();
aa*=ff;
}
int fir[maxn],nxt[2*maxn],to[2*maxn],v[2*maxn],e=0;
void add(int x,int y,int z) {
to[++e]=y;nxt[e]=fir[x];fir[x]=e;v[e]=z;
to[++e]=x;nxt[e]=fir[y];fir[y]=e;v[e]=z;
}
void get_root(int pos,int f) {
size[pos]=1; son[pos]=0; int y,z;
for(y=fir[pos];y;y=nxt[y]) {
if((z=to[y])==f||vis[z]) continue;
get_root(z,pos);
size[pos]+=size[z];
son[pos]=max(son[pos],size[z]);
}
son[pos]=max(son[pos],sumsize-size[pos]);
if(son[pos]<son[root]) root=pos;
}
set<int> G;
int zz[maxn],t;
void get_ans(int x) {
For(i,1,m) if(G.find(p[i]-x)!=G.end()) ans[i]|=1;
}
void dfs(int pos,int f) {
int y,z; zz[++t]=len[pos];
get_ans(len[pos]);
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]||z==f) continue;
len[z]=len[pos]+v[y]; dfs(z,pos);
}
}
void solve(int pos) {
vis[pos]=1; G.clear(); G.insert(0);
int y,z;
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]) continue;
len[z]=v[y]; dfs(z,pos);
while(t) G.insert(zz[t--]);
}
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]) continue;
sumsize=size[z]; root=0;
get_root(z,0);solve(root);
}
}
int main() {
read(n); read(m);
int x,y,z;
For(i,1,n-1) {
read(x); read(y); read(z);
add(x,y,z);
}
For(i,1,m) read(p[i]);
son[root=0]=sumsize=n; get_root(1,0); solve(root);
For(i,1,m) printf(ans[i]? "AYE
":"NAY
");
return 0;
}
bzoj2152聪聪可可
//Serene
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
using namespace std;
#define ll long long
#define db double
#define For(i,a,b) for(int i=(a);i<=(b);++i)
#define Rep(i,a,b) for(int i=(a);i>=(b);--i)
const int maxn=4e4+7;
int n,root,vis[maxn],nowsum;
char cc; ll ff;
template<typename T>void read(T& aa) {
aa=0;cc=getchar();ff=1;
while((cc<'0'||cc>'9')&&cc!='-') cc=getchar();
if(cc=='-') ff=-1,cc=getchar();
while(cc>='0'&&cc<='9') aa=aa*10+cc-'0',cc=getchar();
aa*=ff;
}
ll gcd(ll x,ll y) {
return y==0? x:gcd(y,x%y);
}
int fir[maxn],nxt[2*maxn],to[2*maxn],e=0,v[2*maxn];
void add(int x,int y,int z) {
to[++e]=y;nxt[e]=fir[x];fir[x]=e;v[e]=z%3;
to[++e]=x;nxt[e]=fir[y];fir[y]=e;v[e]=z%3;
}
int son[maxn],size[maxn];
void get_root(int pos,int f) {
size[pos]=1; son[pos]=0; int y,z;
for(y=fir[pos];y;y=nxt[y]) {
if((z=to[y])==f||vis[z]) continue;
get_root(z,pos);
size[pos]+=size[z];
if(size[z]>=son[pos]) son[pos]=size[z];
}
son[pos]=max(son[pos],nowsum-size[pos]);
if(son[pos]<son[root]) root=pos;
}
ll ans;
ll now[10],tot[10],t;
void dfs(int pos,int f,int x) {
++now[x]; int y,z;
ans+=2*tot[(3-x)%3];
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]||z==f) continue;
dfs(z,pos,(x+v[y])%3);
}
}
void solve(int pos) {
vis[pos]=1; int y,z;
memset(tot,0,sizeof(tot)); tot[0]=1;
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]) continue;
dfs(z,pos,v[y]%3);
For(i,0,2) tot[i]+=now[i],now[i]=0;
}
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]) continue;
nowsum=size[z]; root=0;
get_root(z,pos); solve(root);
}
}
int main() {
read(n); int x,y,z; ans=n;
For(i,1,n-1) {
read(x); read(y); read(z);
add(x,y,z);
}
size[0]=son[0]=nowsum=n;
get_root(1,0); solve(root);
ll r=n*n;ll p=gcd(ans,r);
ans/=p; r/=p;
printf("%lld/%lld",ans,r);
return 0;
}
bzoj3648寝室管理
裸的点分治，环的处理就是直接把环跑一遍就可以了。
//Serene
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
using namespace std;
#define ll long long
#define db double
#define For(i,a,b) for(int i=(a);i<=(b);++i)
#define Rep(i,a,b) for(int i=(a);i>=(b);--i)
const int maxn=1e5+7;
int n,m,k,A,B,root,SUM;
ll ans;
char cc; ll ff;
template<typename T>void read(T& aa) {
aa=0;cc=getchar();ff=1;
while((cc<'0'||cc>'9')&&cc!='-') cc=getchar();
if(cc=='-') ff=-1,cc=getchar();
while(cc>='0'&&cc<='9') aa=aa*10+cc-'0',cc=getchar();
aa*=ff;
}
int f[maxn];
int find(int x){return x==f[x]? x:f[x]=find(f[x]);}
int fir[maxn],nxt[2*maxn],to[2*maxn],e=0;
void add(int x,int y) {
to[++e]=y;nxt[e]=fir[x];fir[x]=e;
to[++e]=x;nxt[e]=fir[y];fir[y]=e;
}
int sz[maxn];
void chge(int pos,int x) {
while(pos<=n) {
sz[pos]+=x;
pos+=pos&-pos;
}
}
int q(int pos) {
ll rs=0;
while(pos) {
rs+=sz[pos];
pos-=pos&-pos;
}
return rs;
}
int size[maxn],U[maxn];bool vis[maxn];
void get_root(int pos,int f) {
int y,z;U[pos]=0;size[pos]=1;
for(y=fir[pos];y;y=nxt[y]) {
if((z=to[y])==f||vis[z]) continue;
get_root(z,pos);
size[pos]+=size[z];
U[pos]=max(U[pos],size[z]);
}
U[pos]=max(U[pos],SUM-size[pos]);
if(U[pos]<U[root]) root=pos;
}
int zz[maxn],s,t;
void dfs(int pos,int f,int d,int dd) {//d:push dd:query
zz[++t]=d; int y,z;
ans+=(ll)q(n)-q(max(k-dd-1,0));
for(y=fir[pos];y;y=nxt[y]) {
if((z=to[y])==f||vis[z]) continue;
dfs(z,pos,d+1,dd+1);
}
}
void solve(int pos) {
vis[pos]=1; int y,z;
zz[s=t=1]=1;chge(1,1);
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]) continue;
dfs(z,pos,2,1);
For(i,s+1,t) chge(zz[i],1);
s=t;
}
For(i,1,t) chge(zz[i],-1);
for(y=fir[pos];y;y=nxt[y]) {
if(vis[z=to[y]]) continue;
SUM=size[z]; root=0; get_root(z,pos);
solve(root);
}
}
int h[maxn],tot;
bool get_h(int pos,int f) {
h[++tot]=pos; int y,z;
if(pos==B) return vis[pos]=1;
for(y=fir[pos];y;y=nxt[y]) {
if((z=to[y])==f) continue;
if(get_h(z,pos)) return vis[pos]=1;
}
tot--;
return 0;
}
void get_ans() {
memset(vis,0,sizeof(vis));
tot=0;s=t=0; get_h(A,0);
For(r,1,tot) {
dfs(h[r],0,r,tot-r+1);
For(i,s+1,t) chge(zz[i],1);
s=t;
}
}
int main() {
read(n); read(m); read(k);
int x,y;
For(i,1,n) f[i]=i;
For(i,1,m) {
read(x); read(y);
if(find(x)==find(y)) A=x,B=y;
else {
f[find(x)]=find(y);
add(x,y);
}
}
U[0]=n+1; SUM=n;
get_root(1,0); solve(root);
if(n==m) get_ans();
printf("%lld",ans);
return 0;
}
/*
5 5 2
1 3
2 4
3 5
4 1
5 2
*/