csu1798 小Z的城市

原创 2016年08月30日 15:28:46

小Z的城市

Time Limit: 5 Sec  Memory Limit: 128 MB

Description

小Z身为A国的君王,A国的所有城市都归小A所有,A国中从任意一个城市出发都能到达其他任意一个城市,每两个直接相邻的城市xi , yi之间都有一条已知长度的路径相连。A国共有n个城市(编号为1~n),被n-1条无向边相连。小Y和小X的开国功臣,但是小X擅文,小Y好武,所以他们关系很差,现在小Z想在区间[l ,r]的城市中选择两个奖赏这两位开国功臣一人一个城市,彼此厌恶的小X和小Y想让聪明的你飞到A国帮助他们做出选择可以距离彼此更远。

(也就是说最后小X会选择城市idx(l<=idx<=r) , 小Y会选择城市idy(l<=idy<=r) , 保证idx!=idy 而且idx距离idy尽可能远)。

Input

多组数据

每组数据第一行两个正整数n和q 表示A国有n个城市,q次询问 ( 2<=n <=100000 , q<=100000 )

接下来n-1行,每行输入3个正整数xi , yi , di 表示城市xi与yi被一条长度为di的路径相连

(1<=xi , yi<=n , di<=100)

接下来q个询问,每次询问输入一行li和ri,对于每次询问希望在li和ri区间选择两个城市分给小X和小Y,使他们距离最大 (1<=li < ri<=n)

所有数据保证Σn <=500000

Output

每次询问输出一个使小X和小Y距离的最大值

Sample Input

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

Sample Output

10


可以参考http://blog.csdn.net/werkeytom_ftd/article/details/51791519

其实这个问题可以转换为每次给你一个子树,求子树的直径,如果给的点是不联通的,加入中间的点使他联通,这不影响原本的最远点对。

然后树的直径有这么一个性质,两颗树合成一棵树,新树的最远点对一定是原本两棵树最远点对中的点。你可以想象一下一棵树随便找个地方切一刀变成两棵树,那么原本的最远点对在新的两棵树中肯定也是最远的点对最佳候补,大概这么一个感觉。

然后这题就可以用线段树维护查询了,合并的时候只要把左右区间的最远点对取出来枚举一下哪两个可以形成新的最远点对。至于两点距离,可以先预处理出每个点到根结点的距离dis[i],然后两点u和v的距离就是dis[u]+dis[v]-dis[lca(u,v)],lca(u,v)为两点公共祖先。


以下代码

#include <stdio.h>
#include <algorithm>
#include <set>
#include <string.h>
#include <math.h>
#include <iostream>
#include <vector>
#include <map>
#include <queue>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const double PI = acos(-1.0);
const ll inf = 0x3f3f3f3f3f3f3f3fll;
const int mod = 1e9 + 7;
const int maxn = 100000 + 10;

//顶点编号从0开始
const int max_v = 222222;
const int max_log_v = 22;

vector<PII> g[max_v];
int root;

int parent[max_log_v][max_v];
int depth[max_v];
int cost[max_v];

void dfs(int v, int p, int d, int c) {
    parent[0][v] = p;
    depth[v] = d;
    cost[v] = c;
    for (int i = 0; i < g[v].size(); ++i) {
        if (g[v][i].fi != p) dfs(g[v][i].fi, v, d + 1, c + g[v][i].se);
    }
}

void init(int V) {
    dfs(root, -1, 0, 0);
    for (int k = 0; k + 1 < max_log_v; ++k) {
        for (int v = 0; v < V; ++v) {
            if (parent[k][v] < 0) parent[k + 1][v] = -1;
            else parent[k + 1][v] = parent[k][parent[k][v]];
        }
    }
}

int lca(int u, int v) {
    if (depth[u] > depth[v]) swap(u, v);
    for (int k = 0; k < max_log_v; ++k) {
        if ((depth[v] - depth[u]) >> k & 1) {
            v = parent[k][v];
        }
    }
    
    if (u == v) return u;
    
    for (int k = max_log_v - 1; k >= 0; k--) {
        if (parent[k][u] != parent[k][v]) {
            u = parent[k][u];
            v = parent[k][v];
        }
    }
    return parent[0][u];
}

int get_distance(int u, int v) {
    return cost[u] + cost[v] - 2 * cost[lca(u, v)];
}



vector<int> point[4 * maxn];

vector<int> get_point(vector<int> a, vector<int> b) {
    for (int i = 0; i < b.size(); i++) {
        a.push_back(b[i]);
    }
    if (a.size() <= 2) return a;
    
    int mx = 0;
    vector<int> ans;
    for (int i = 0; i < a.size(); i++) {
        for (int j = i + 1; j < a.size(); ++j) {
            int dis = get_distance(a[i], a[j]);
            if (dis > mx) {
                mx = dis;
                ans.clear();
                ans.push_back(a[i]);
                ans.push_back(a[j]);
            }
        }
    }
    return ans;
}

void get_tree(int k, int l, int r) {
    if (r - l == 1) {
        point[k].push_back(l);
        return;
    }
    
    get_tree(k * 2 + 1, l, (l + r) / 2);
    get_tree(k * 2 + 2, (l + r) / 2, r);
    point[k] = get_point(point[k * 2 + 1], point[k * 2 + 2]);
}

vector<int> query(int a, int b, int k, int l, int r) {
    vector<int> ans;
    if (r <= a || b <= l) return ans;
    
    if (a <= l && r <= b) return point[k];
    else {
        vector<int> vl = query(a, b, k * 2 + 1, l, (l + r) / 2);
        vector<int> vr = query(a, b, k * 2 + 2, (l + r) / 2, r);
        return get_point(vl, vr);
    }
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("test.in", "r", stdin);
#endif
    
    int n, q;
    while (~scanf("%d%d", &n, &q)) {
        for (int i = 0; i < n; i++) g[i].clear();
        for (int i = 0; i < 4 * n; i++) point[i].clear();
        
        for (int i = 0; i < n - 1; i++) {
            int u, v, c;
            scanf("%d%d%d", &u, &v, &c);
            u--, v--;
            g[u].push_back(PII(v, c));
            g[v].push_back(PII(u, c));
        }
        root = 0;
        init(n);
        get_tree(0, 0, n);
        
        
        for (int i = 0; i < q; ++i) {
            int l, r;
            scanf("%d%d", &l, &r);
            l--, r--;
            vector<int> v = query(l, r + 1, 0, 0, n);
            printf("%d\n", get_distance(v[0], v[1]));
        }
    }
}


版权声明:本文为博主原创文章,未经博主允许不得转载。

csu1798: 小Z的城市 线段树+ST表

题目链接点这里 终于把去年省热身赛的这题补了,,,好久没写这么复杂的题了,,写了2个小时,,调了一个小时bug。。总共花3小时,,,还是太菜了。。 ,,大概,用ST表求任意2点间的距离 ...
  • qq_30927651
  • qq_30927651
  • 2017年04月04日 22:07
  • 158

【BZOJ2038】小Z的袜子,第一次的莫队算法

传送门 写在前面:莫队竟如此暴力…… 思路:当初我对这个题的第一感觉——这个区间问题可以用线段树或者树状数组?答案当然是不能,于是我就去简单学了下莫队算法。在我看来,莫队(分块版,不是二维曼哈顿距...
  • xym_CSDN
  • xym_CSDN
  • 2016年03月14日 19:59
  • 1748

006 - 微信小程序开发之城市选择器 城市切换

移动开发中城市选择器必不可少. 空白造了个. 修改了一下,看这个.之前这个有问题 gif: 这里只上部分js代码: var city = require('../../util...
  • liguanjie8
  • liguanjie8
  • 2017年01月23日 13:13
  • 629

微信小程序获取当前所在城市

本篇文章主要讲解在微信小程序中,如何利用微信自带的api(wx.getLocation())结合百度地图的逆地址解析api来获取当前所在城市名。 实现起来也比较简单,步骤为: 1--利用微信小程序...
  • nongweiyilady
  • nongweiyilady
  • 2017年07月04日 12:26
  • 5285

微信小程序实用组件:城市切换

http://www.wxapp-union.com/forum.php?mod=viewthread&tid=1644&extra=page%3D1%26filter%3Dlastpost%26or...
  • sinat_17775997
  • sinat_17775997
  • 2017年03月12日 12:55
  • 466

[微信小程序]获取用户当前的城市

获取用户当前的城市
  • qq_35713752
  • qq_35713752
  • 2017年11月06日 15:09
  • 372

微信小程序 仿美团城市选择 城市切换

原来的城市选择做了一些修改.代码很简单.var city = require('../../utils/city.js'); var app = getApp() Page({ data: { ...
  • qq_31383345
  • qq_31383345
  • 2017年01月14日 15:25
  • 8986

莫队总结&bzoj 2038 小Z的袜子

莫队算法入门
  • unicornt_
  • unicornt_
  • 2016年07月07日 08:24
  • 318

tsinsen-A1206 小Z的袜子(莫队算法)

题目: http://www.tsinsen.com/A1206 题意: 有许多袜子,每个有自己的颜色,排成一排。给出许多区间询问,求每个区间中任取两个袜子的颜色相同的概率,分数表示。 思路: 区间...
  • kopyh
  • kopyh
  • 2016年04月08日 13:13
  • 388

微信小程序中根据字母选择城市

http://www.wxapp-union.com/article-2774-1.html?utm_source=QQqun 今天开发一个小程序,里面涉及到区域选择,看了网...
  • llixiangjian
  • llixiangjian
  • 2017年07月28日 09:56
  • 2211
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:csu1798 小Z的城市
举报原因:
原因补充:

(最多只允许输入30个字)