codeforces-213 G-Path Queries(そして検索セット+オフライン)

テーマリンク:http://codeforces.com/problemset/problem/1213/G
まず、2つの数n、m（1<=n、m<=2 e 5）を与え、木の数と問い合わせ木を表し、次にn-1行、各行の3つの数u、v、w
ツリー上のuからvノードにはwのエッジがあり、次の行のm個の問い合わせがあり、各問い合わせはツリー上の2つのセクションを出力することを要求する。
点（u，v）（u＜v）（u、v）（u＜v）（u、v）（u＜v）の間の最短距離の中で最も長い方の辺はm＿i m_より小さい。iミの対数
考え方：まずお聞きしたいのですが、小さい時から大きな順に並べ替えて、オフライン処理します。最初の問い合わせより小さいのは間違いなく第二の問い合わせより小さいです。
ある辺は権力の値で並べ替えて、毎回検索する値m i m i m_に対してiミ、辺の権利値をm i m_に等しい値以下にする。i miのポイントを使って、まとめて調べる。
連結ブロックのすべての点が条件を満たしています。点数で貢献を計算して、最後に計算すればいいです。

#include 
#include  
#include   
#include   
#include    
#include    
#include     
#include     
#include     
#include       
#include       
#define dll(x) scanf("%I64d",&x)
#define xll(x) printf("%I64d
",x)
#define pii pair
#define pll pair
#define gb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define mem(X) memset((X), 0, sizeof((X)))
#define memc(X) memset((X), '\0', sizeof((X)))
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define eps 1e-6
#define gg(x) getInt(&x)
using namespace std;
typedef long long ll;
ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;}
ll lcm(ll a, ll b) {return a / gcd(a, b) * b;}
ll powmod(ll a, ll b, ll MOD) {ll ans = 1; while (b) {if (b % 2)ans = ans * a % MOD; a = a * a % MOD; b /= 2;} return ans;}
inline void getInt(int* p);
const int inf = 0x3f3f3f3f;
const int N = 2e5 + 10;

ll ans, sum[N], res[N];
int n, m, pre[N];
struct node {
	int u, v, w;
} e[N];

struct query {
	int x, idx;
} q[N];

ll cal(ll x) {return x * (x - 1) / 2;}
int find(int x) {return pre[x] == x ? x : pre[x] = find(pre[x]);}

void add(int a, int b)
{
	int fa = find(a);
	int fb = find(b);
	if (sum[fa] > sum[fb])swap(fa, fb);
	pre[fa] = fb;
	ans -= cal(sum[fa]); ans -= cal(sum[fb]);
	sum[fb] += sum[fa];
	ans += cal(sum[fb]);
}

int main(int argc, char const *argv[])
{
	gb;
#ifdef ONLINE_JUDGE
#else
	freopen("D:\\test\\in.txt", "r+", stdin);
#endif
	cin >> n >> m;
	for (int i = 0; i <= n; i++)pre[i] = i, sum[i] = 1;
	for (int i = 1; i < n; i++) {
		cin >> e[i].u >> e[i].v >> e[i].w;
	}
	sort(e + 1, e + n, [](node a, node b) {return a.w < b.w;});
	for (int i = 1; i <= m; i++) {
		cin >> q[i].x;
		q[i].idx = i;
	}
	sort(q + 1, q + 1 + m, [](query a, query b) {return a.x < b.x;});
	int now = 1;
	for (int i = 1; i <= m; i++) {
		int tmp = q[i].x;
		while (now < n && e[now].w <= tmp) {
			add(e[now].u, e[now].v);
			now++;
		}
		res[q[i].idx] = ans;
	}
	for (int i = 1; i <= m; i++) {
		cout << res[i] << " ";
	}
	return 0;
}