POJ 1949 [最小优先队列]

　　题意是有N行任务,给出完成每项任务所需要的时间长度,求出完成所有任务所需要的最短时间.每个任务都会有一个约束条件,就是在完成这项任务之前必须完成所列出的其它任务.可以同时做多项任务.简单来说就像煮饭炒菜问题一样,可以一边烧饭一边炒菜.但炒菜之前必先洗菜.

　　这道题我主要用了最小优先队列,有点像dijktra, 列队按两个条件来维护, 首先是所按所需的条件(前任务)个数,这里用Need[]数组保存,然后按开始任务的时间. 用一个Need[]数组维护每个任务所需的前任务个数,如果为0, 则丢进列队里, 每次取出列队首元素, 即当前可以开始的任务, 将所有以这一任务为前提的任务的Need减一,再更新一下start值(表示任务开始的时间). 若Need为0,则丢进列队里,如此重复直到列队为空, 再找出start里最大值,再加上这个最大start值的任务的所需时间,即为完成所有任务所需的最短时间.

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>

 using namespace std;

const int MAXN=10001;
const int INF=0x7f7f7f7f;

vector<int> Box[MAXN];
int Cost[MAXN], Start[MAXN], Need[MAXN];

void init_input(const int n)
{
    int num, temp;
    memset(Start, 0x7f, sizeof(Start));
    //printf("%d\n", Start[0]);
    for( int i=0; i<n; ++i )
    {
        scanf("%d%d", &Cost[i], &num);
        //cin>>Cost[i]>>num;
        Need[i]=num;
        while( num-- )
        {
            scanf("%d", &temp);
            Box[temp-1].push_back(i);
        }
    }
}
void clear(const int n)
{
    for( int i=0; i<n; ++i )
        Box[i].clear();
}

class comp
{
    public:
        bool operator()(const int i, const int j)
        {
            if( Need[i] == Need[j] )
                return Start[i] > Start[j];
            else
                return Need[i] > Need[j];
        }
};

int solve(const int n)
{
    priority_queue<int, vector<int>, comp> que;
    for( int i=0; i<n; ++i )
    {
        if( Need[i] == 0 )
        {
            Start[i]=0;
            que.push(i);
        }
    }
    int ans=0;
    while( !que.empty() )
    {
        int front=que.top();
        que.pop();
        //printf("%d\n", front);
        vector<int>::iterator ix=Box[front].begin();
        while( ix != Box[front].end() )
        {
            Need[*ix]--;
            //printf("%d\n", *ix);
            if( Start[*ix] == INF || Start[*ix] < Start[front]+Cost[front] )
            {
                Start[*ix] = Start[front]+Cost[front];
            }
            if( Need[*ix] == 0 )
                que.push(*ix);
            ++ix;
        }
    }
    for( int i=0; i<n; ++i )
    {
        ans=Start[i]+Cost[i]>ans?Start[i]+Cost[i]:ans;
    }
    return ans;
}

int main()
{
    freopen("in", "r", stdin);
    freopen("out", "w", stdout);

    int n;
    while( cin>>n )
    {
        init_input(n);
        printf("%d\n", solve(n));
        clear(n);
    }
    return 0;
}