BZOJ 3648: 寝室管理( 点分治 + 树状数组 )

1棵树的话, 点分治+你喜欢的数据结构(树状数组/线段树/平衡树)就可以秒掉, O(N log^2 N). 假如是环套树, 先去掉环上1条边, 然后O(N log^2 N)处理树(同上); 然后再O(N log N)计算经过删掉边的路径数(在环上扫一遍, 数据结构维护).

-------------------------------------------------------------------------

#include<cstdio>
#include<cctype>
#include<cstring>
#include<iostream>
#include<algorithm>
 
using namespace std;
 
typedef long long ll;
 
const int maxn = 100009;
 
int N, M, K, e_u, e_v, n, Rt, T, _T, len;
int par[maxn], sz[maxn], mx[maxn], nxt[maxn];
bool vis[maxn];
ll ans;
 
template<class T>
inline void Max(T &x, T t) {
if(t > x) x = t;
}
 
inline int getint() {
char c = getchar();
for(; !isdigit(c); c = getchar());
int ret = 0;
for(; isdigit(c); c = getchar())
ret = ret * 10 + c - '0';
return ret;
}
 
struct edge {
int t;
edge* n;
} E[maxn << 1], *pt = E, *H[maxn];
 
inline void AddEdge(int u, int v) {
pt->t = v, pt->n = H[u], H[u] = pt++;
}
 
void dfs_rt(int x, int fa = -1) {
mx[x] = sz[x] = 1;
for(edge* e = H[x]; e; e = e->n) if(e->t != fa && !vis[e->t]) {
dfs_rt(e->t, x);
sz[x] += sz[e->t];
Max(mx[x], sz[e->t]);
}
Max(mx[x], n - sz[x]);
if(Rt < 0 || mx[x] < mx[Rt]) Rt = x;
}
 
void dfs_sz(int x, int fa = -1) {
sz[x] = 1;
for(edge* e = H[x]; e; e = e->n) if(e->t != fa && !vis[e->t]) {
dfs_sz(e->t, x);
sz[x] += sz[e->t];
}
}
 
struct BIT {
int B[maxn], mk[maxn];
BIT() {
memset(B, 0, sizeof B);
}
inline void Add(int p, int c) {
for(; p <= N; p += p & -p) if(mk[p] != c) {
B[p] = 1;
mk[p] = c;
} else
B[p]++;
}
inline int Sum(int p, int c) {
if(p < 0) return 0;
int ret = 0;
for(; p; p -= p & -p)
if(mk[p] == c) ret += B[p];
return ret;
}
inline int Query(int l, int r, int c) {
return Sum(r, c) - Sum(l - 1, c);
}
} A, B, *CUR[2];
 
void dfs_add(int v, int t, int x, int d, int fa = -1) {
CUR[v]->Add(d++, t);
for(edge* e = H[x]; e; e = e->n)
if(e->t != fa && !vis[e->t]) dfs_add(v, t, e->t, d, x);
}
 
void Solve(int x) {
Rt = -1;
dfs_rt(x);
vis[x = Rt] = true;
dfs_sz(x);
for(edge* e = H[x]; e; e = e->n) if(!vis[e->t]) {
dfs_add(0, ++T, e->t, 1);
for(int i = 1; i <= sz[e->t]; i++)
ans += ll(CUR[0]->Query(i, i, T)) * CUR[1]->Query(K - i, N, Rt);
dfs_add(1, Rt, e->t, 2);
}
ans += CUR[1]->Query(K, N, Rt);
for(edge* e = H[x]; e; e = e->n) if(!vis[e->t]) {
n = sz[e->t];
Solve(e->t);
}
}
 
bool DFS_C(int x, int fa = -1) {
for(edge* e = H[x]; e; e = e->n) if(e->t != fa) {
nxt[e->t] = x;
if(e->t == e_v) {
len = 2;
return true;
}
if(DFS_C(e->t, x)) {
len++;
return true;
}
}
return false;
}
 

void calc(int x, int v, int d) {

vis[x] = true;
_T++;
dfs_add(v, _T, x, d, nxt[x]);
dfs_sz(x, nxt[x]);
for(int i = d; i <= d + sz[x]; i++)
ans += ll(CUR[v]->Query(i, i, _T)) * (CUR[v ^ 1]->Query(K - i, N, T));
dfs_add(v ^ 1, T, x, len - d + 1, nxt[x]);
if(!vis[nxt[x]])
calc(nxt[x], v, d - 1);
}
 
void Work() {
CUR[0] = &A, CUR[1] = &B;
memset(vis, 0, sizeof vis);
n = N;
Solve(T = 0);
if(M < N) {
cout << ans << endl;
return;
}
A = BIT(), B = BIT();
CUR[0] = &A, CUR[1] = &B;
T = _T = 0;
DFS_C(e_u);
nxt[e_u] = e_v;
++T;
memset(vis, 0, sizeof vis);
vis[e_v] = true;
dfs_add(0, T, e_v, 1, nxt[e_v]);
calc(nxt[e_v], 1, len - 1);
cout << ans << "\n";
}
 
int Find(int x) {
return x == par[x] ? x : par[x] = Find(par[x]);
}
 
void Init() {
N = getint(), M = getint(), K = getint();
for(int i = 0; i < N; i++) par[i] = i;
for(int i = 0; i < M; i++) {
int u = getint() - 1, v = getint() - 1;
int _u = Find(u), _v = Find(v);
if(_u != _v) {
par[_u] = _v;
AddEdge(u, v);
AddEdge(v, u);
} else
e_u = u, e_v = v;
}
ans = 0;
}
 
int main() {
Init();
Work();
return 0;
}

-------------------------------------------------------------------------

3648: 寝室管理

Time Limit: 40 Sec  Memory Limit: 512 MB
Submit: 136  Solved: 67
[Submit][Status][Discuss]

Description

    T64有一个好朋友,叫T128。T128是寄宿生,并且最近被老师叫过去当宿管了。宿管可不是一件很好做的工作,碰巧T128有一个工作上的问题想请T64帮忙解决。
  T128的寝室条件不是很好,所以没有很多钱来装修。礼间寝室仅由n-1条双向道路连接,而且任意两间寝室之间都可以互达。最近,T128被要求对一条路径上的所有寝室进行管理,这条路径不会重复经过某个点或某条边。但他不记得是哪条路径了。他只记得这条路径上有不少于k个寝室。于是,他想请T64帮忙数一下,有多少条这样的路径满足条件。
    嗯…还有一个问题。由于最近有一些熊孩子不准晚上讲话很不爽,他们决定修筑一条“情报通道”,如果通道建成,寝室就变成了一个N个点N条边的无向图。并且,经过“情报通道”的路径也是合法的。T128心想:通道建成之前,T64还有一个高效的算法帮我数路径条数,但是通道建成之后,他还有办法吗?对,T64手忙脚乱,根本数不清有多少条路径。于是他找到了你。

Input

   第一行为三个正整数N,M,K(2 ≤ K ≤ N),代表有n间寝室,m条边连接它们n-1 ≤ m ≤ N;m= n-1意味着“情报遁道”未被修好;m=n意味着“情报通道”已被修好),以及题目描述中的K。
  接下来m行,每行两个正整数z,y,代表第x间寝室与第y间寝室之间有一条双向边。

Output

  
仅包含一个整数,代表经过至少K间寝室的路径条数。

Sample Input

5 5
1 3
2 4
3 5
4 1
5 2

Sample Output

20

HINT



N≤100000      

K≤N

M=N           

Source

 

posted @ 2016-02-13 16:23  JSZX11556  阅读(525)  评论(0编辑  收藏