SCOI2013 day1 摩托车交易(motorcycle)

题目描述：

先吐槽一下为什么今年题要用纸质的…………又是版权问题么…………

然后数据因为版权…………没法得到，所以现在只能在电子科大OJ上提交题目（真搞不懂，数据什么时候也不公开了…………）。

 

首先很明显这道题要求我们找两点之间最小边的最大值，然后能买尽量买，能卖尽量卖，贪心走一遍就好了。考试的时候没有想出来正确的解法，于是上了一个类似弗洛伊德的DP，悲剧的是忘了处理列车站的情况，然后中间也是各种差错导致挂零…………悲剧的人生啊……

其实这个问题很简单，两点之间的最小边的最大值一定存在于最大生成树上（因为最大生成树上两点之间的路径是所有路径中包含的边中权值最大的几条），然后就成了在最大生成树上倍增了，此题瞬间变水题……无语啊……为什么我这么弱……………………T T

需要注意的：1.有列车站的城市需要缩成一个点来做。

　　　　　　2.中途的黄金有可能超过int，因此在倍增的时候返回INF的话一定要返回long long的INF。

我的代码，求神牛们轻虐……

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
#define min(a,b) ((a)<(b)?(a):(b))
#define max(a,b) ((a)>(b)?(a):(b))
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int MAXN=100010;
const int MAXM=500010;

struct Edge
{
    int st,ed,val;
    bool operator < (const Edge &a) const
    {
        return val>a.val;
    }
}edge[MAXM];
struct Node
{
    int end,val;
    Node *next;
}edge2[MAXN<<1],*head[MAXN];

int n,m,q,all,all2,gold[MAXN],ord[MAXN],lb[MAXN];
int set_fa[MAXN],fa[MAXN][22],dis[MAXN][22],dep[MAXN];
int que[MAXN];

int find_set(int a)
{
    if (set_fa[a]==a) return a;
    return set_fa[a]=find_set(set_fa[a]);
}
void union_set(int a,int b)
{
    int f1=find_set(a);
    int f2=find_set(b);
    if (f1!=f2) set_fa[f1]=f2;
}
void add_edge(int a,int b,int v)
{
    edge[all].st=a,edge[all].ed=b,edge[all++].val=v;
}
void add_tree_edge(int a,int b,int v)
{
    edge2[all2].end=b,edge2[all2].val=v,edge2[all2].next=head[a],head[a]=edge2+(all2++);
}
void make_tree()
{
    int choose=0,all=-1;
    if (!q) q=1;
    while (choose<n-q)
    {
        ++all;
        if (find_set(edge[all].st)!=find_set(edge[all].ed))
        {
            ++choose;
            add_tree_edge(edge[all].st,edge[all].ed,edge[all].val);
            add_tree_edge(edge[all].ed,edge[all].st,edge[all].val);
            union_set(edge[all].st,edge[all].ed);
        }
    }
    int front=0,rear=1,t;
    dep[que[rear]=(lb[1]?n+1:1)]=1;
    while (front!=rear)
    {
        t=que[++front];
        for (Node *p=head[t];p;p=p->next)
            if (p->end!=fa[t][0])
            {
                fa[p->end][0]=t;
                dis[p->end][0]=p->val;
                dep[p->end]=dep[t]+1;
                que[++rear]=p->end;
            }
    }
    for (int k=1;k<=20;++k)
        for (int i=1;i<=n+1;++i)
        {
            if (lb[i]) continue;
            fa[i][k]=fa[fa[i][k-1]][k-1];
            dis[i][k]=min(dis[i][k-1],dis[fa[i][k-1]][k-1]);
        }
}
ll get_val(int a,int b)
{
    if (lb[a]) a=n+1;
    if (lb[b]) b=n+1;
    if (a==b) return INF;
    int i;ll ans=INF;
    if (dep[a]<dep[b]) swap(a,b);
    for (i=20;~i;--i) if (dep[fa[a][i]]>=dep[b]) ans=min(ans,dis[a][i]),a=fa[a][i];
    if (a==b) return ans;
    for (i=20;~i;--i) if (fa[a][i]!=fa[b][i]) ans=min(ans,min(dis[a][i],dis[b][i])),a=fa[a][i],b=fa[b][i];
    return min(ans,min(dis[a][0],dis[b][0]));
}
int main()
{
    freopen("motorcycle.in","r",stdin);
    freopen("motorcycle.out","w",stdout);
    int x,y,z;
    scanf("%d%d%d",&n,&m,&q);
    for (int i=1;i<=n+1;++i)
        set_fa[i]=i;
    for (int i=1;i<=n;++i)
        scanf("%d",&ord[i]);
    for (int i=1;i<=n;++i)
        scanf("%d",&gold[i]);
    while (m--)
    {
        scanf("%d%d%d",&x,&y,&z);
        add_edge(x,y,z);
    }
    for (int i=1;i<=q;++i)
    {
        scanf("%d",&x);
        lb[x]=1;
    }
    for (int i=0;i<all;++i)
    {
        if (lb[edge[i].st]) edge[i].st=n+1;
        if (lb[edge[i].ed]) edge[i].ed=n+1;
    }
    sort(edge,edge+all);
    make_tree();
    ll now=max(gold[ord[1]],0),now_v;
    if (gold[ord[1]]<0) puts("0");
    for (int i=2;i<=n;++i)
    {
        now_v=get_val(ord[i-1],ord[i]);
        if (gold[ord[i]]>0)
            now=min(now,now_v)+gold[ord[i]];
        else 
        {
            now=min(now,now_v);
            printf("%lld\n",min(-gold[ord[i]],now));
            now=max(now+gold[ord[i]],0);
        }
    }
    return 0;
}