【vijos1780】【NOIP2012】开车旅行 倍增

题目描述

　　有$n$个城市，第$i$个城市的海拔为$h_i$且这$n$个城市的海拔互不相同。编号比较大的城市在东边。两个城市$i,j$之间的距离为$|h_i-h_j|$

　　小A和小B要开车去旅行。小A先开，他们会轮流开车。小A会把车开到第二近的城市，小B会把车开到最近的城市。如果当前城市到两个城市的距离相同，则认为海拔低的城市比较近。他们只会把车往东边开（即编号大的那边）。

　　小A会先问你对于一个给定的$x=x_0$，从哪一个城市出发，小A开车行驶的路程总数与小B行驶的路程总数的比值最小（如果小B的行驶路程为$0$，此时的比值可视为无穷大，且两个无穷大视为相等）。如果从多个城市出发，小A开车行驶的路程总数与小B行驶的路程总数的比值都最小，则输出海拔最高的那个城市。

　　他还会问你$m$次。每次要你求出对给定的$x=x_i$ 和出发城市$s_i$，小A开车行驶的路程总数以及小B行驶的路程总数。

　　$n\leq 100000,m\leq 10000$

题解

　　先用set求出当前在第$i$个城市时小A会把车开到哪个城市和小B会把车开到那个城市。

　　然后倍增。$fa_{i,j}$表示当前在第$i$个城市时开$2^j$轮（小A和小B各开一次算一轮）后小A开车的距离，$f_b{i,j}$表示当前在第$i$个城市时开$2^j$轮后小B开车的距离。$g_{i,j}$表示当前在第$i$个城市时开$2^j$轮后会到达哪里。

　　现在我们要求当前在第$s$个城市，开不超过$x$公里，小A开车的路程和小B开车的路程。显然直接倍增就可以了，还要加上不满一轮的情况，即开若干轮后小A再开一次。

　　第一问直接枚举起点，计算答案。

　　第二问直接计算答案。

　　时间复杂度：$O((n+m)logn)$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#include<utility>
#include<set>
#include<cmath>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,int> pli;
ll h[100010];
set<pli> ht;
int na[100010];
int nb[100010];
ll fa[100010][20];
ll fb[100010][20];
int g[100010][20];
int getmin(ll v)
{
    if(ht.size()==0)
        return -1;
    set<pli>::iterator p=ht.lower_bound(pli(v,0));
    if(p==ht.begin())
        return p->second;
    if(p==ht.end())
    {
        p--;
        return p->second;
    }
    pii s1=*p;
    s1.first-=v;
    p--;
    pii s2=*p;
    s2.first=v-s2.first;
    if(s1.first<s2.first)
        return s1.second;
    return s2.second;
}
struct p
{
    int x;
    double s;
};
double inf=1e100;
int main()
{
    freopen("vijos1780.in","r",stdin);
    freopen("vijos1780.out","w",stdout);
    int n;
    scanf("%d",&n);
    int i,j;
    for(i=1;i<=n;i++)
        scanf("%lld",&h[i]);
    for(i=n;i>=1;i--)
    {
        nb[i]=getmin(h[i]);
        if(nb[i]==-1)
            na[i]=-1;
        else
        {
            ht.erase(ht.find(pii(h[nb[i]],nb[i])));
            na[i]=getmin(h[i]);
            ht.insert(pii(h[nb[i]],nb[i]));
        }
        ht.insert(pii(h[i],i));
    }
    for(i=1;i<=n;i++)
        if(na[i]==-1)
        {
            g[i][0]=i;
            fa[i][0]=fb[i][0]=0;
        }
        else if(nb[na[i]]==-1)
        {
            g[i][0]=na[i];
            fa[i][0]=abs(h[i]-h[na[i]]);
            fb[i][0]=0;
        }
        else
        {
            g[i][0]=nb[na[i]];
            fa[i][0]=abs(h[i]-h[na[i]]);
            fb[i][0]=abs(h[na[i]]-h[nb[na[i]]]);
        }
    for(j=1;j<=19;j++)
        for(i=1;i<=n;i++)
        {
            g[i][j]=g[g[i][j-1]][j-1];
            fa[i][j]=fa[i][j-1]+fa[g[i][j-1]][j-1];
            fb[i][j]=fb[i][j-1]+fb[g[i][j-1]][j-1];
        }
    int k;
    ll x0;
    scanf("%lld",&x0);
    p ans;
    ans.x=-1;
    for(i=1;i<=n;i++)
    {
        ll sa=0,sb=0;
        j=i;
        ll s=0;
        for(k=19;k>=0;k--)
            if(fa[j][k]+fb[j][k]+s<=x0)
            {
                s+=fa[j][k]+fb[j][k];
                sa+=fa[j][k];
                sb+=fb[j][k];
                j=g[j][k];
            }
        if(na[j]!=-1&&s+abs(h[j]-h[na[j]])<=x0)
            sa+=abs(h[j]-h[na[j]]);
        double si;
        if(!sb)
            si=inf;
        else
            si=double(sa)/sb;
        if(ans.x==-1||si<ans.s||(si==ans.s&&h[i]>h[ans.x]))
        {
            ans.x=i;
            ans.s=si;
        }
    }
    printf("%d\n",ans.x);
    int m;
    scanf("%d",&m);
    for(i=1;i<=m;i++)
    {
        int x;
        ll x0;
        scanf("%d%lld",&x,&x0);
        j=x;
        ll s=0;
        ll sa=0,sb=0;
        for(k=19;k>=0;k--)
            if(fa[j][k]+fb[j][k]+s<=x0)
            {
                s+=fa[j][k]+fb[j][k];
                sa+=fa[j][k];
                sb+=fb[j][k];
                j=g[j][k];
            }
        if(na[j]!=-1&&s+abs(h[j]-h[na[j]])<=x0)
            sa+=abs(h[j]-h[na[j]]);
        printf("%lld %lld\n",sa,sb);
    }
    return 0;
}