HDU 2647 Reward

大体思路就是简单的拓扑排序。

写的时候没有看模板，都是自己一点一点改的，应该没有模板效率高。

还用了一点优先队列的思想，用了并查集的方法检查是否存在环。

#include<cstdio>
#include<cstdlib>
#include<cstring>

using namespace std;

struct N
{
    int data;
    int n,money;
    N *next;
}*head[10010];

void init(int n)//初始化函数
{
    for(int i = 1;i <= n; i++)
    {
        head[i] = (struct N *)malloc(sizeof(struct N));
        head[i]->n = 0;
        head[i]->data = -1;
        head[i]->next = NULL;
        head[i]->money = -1;
    }
}

int h[10010];//为并查集准备的数组

int seek(int u,int v)//并查集
{
    while(h[v] != v)
    {
        v = h[v];
    }
    if(u != v) {h[u] = v;return 1;}
    else return 0;
}

void link(int u,int v)
{
    N *p1 = head[u],*p2 = head[u]->next;
    for(;p2 != NULL; p1 = p1->next,p2 = p2->next)
    {
        if(p2->data == v) return;
        if(p1->data < v && v < p2->data)
        {
            N *p = (struct N *)malloc(sizeof(struct N));
            p->data = v;
            head[v]->n++;
            p->next = p1->next;
            p1->next = p;
            return;
        }
    }
    N *p = (struct N *)malloc(sizeof(struct N));
    p->data = v;
    head[v]->n++;
    p->next = NULL;
    p1->next = p;
    return;
}

int sum;

void check(int u,int min)
{
    int q[10001],s = 0,e = 0;
    for(N *p = head[u]->next; p != NULL; p = p->next)
    {
        if(head[p->data]->money == -1)//如果此点第一次被访问，直接将min加入sum
        {
            sum += min;
            head[p->data]->money = min;
        }
        else if(head[p->data]->money < min)//如果此点已经被访问，则需比较当前的min是否大于其当前的奖励。
        {
            sum = sum + (min - head[p->data]->money);
            head[p->data]->money = min;
        }
        q[e++] = p->data;
    }
    for(s = 0,min = min+1;s < e; s++)
    {
        check(q[s],min);
    }
}

int main()
{

    int m,n,u,v,i,flag;
    while(scanf("%d %d",&n,&m) != EOF)
    {
        for(i = 1;i <= n; i++)
            h[i] = i;
        init(n);
        flag = 1;
        while(m--)
        {
            scanf("%d %d",&v,&u);
            if(flag && seek(u,v))//seek——并查集  用来检查是否存在环
            {
                link(u,v);
            }
            else flag = 0;
        }
        if(flag)
        {
            sum = 888*n;
            for(i = 1;i <= n; i++)
            {
                if(head[i]->n == 0)
                    check(i,1);
            }
            printf("%d\n",sum);
        }
        else printf("-1\n");
    }
    return 0;
}