中兴 软件开发 笔试

int calculateMaxOil(int numNodes, int sourceNode, vector<vector<int> > & network)
{
    // WRITE YOUR CODE HERE
    long long head[101010];
    long long sum[101010];
    int maxn;
    vector<int> a;
    vector<int> b;
    maxn=1111111;
    
    for (int i=0;i<=numNodes+1;i++) 
    {
        head[i]=maxn;
        sum[i]=0;
    }
    
    for (vector<vector<int> >::iterator it=network.begin();it!=network.end();it++)
    {
        a=*it;
        head[a[1]]=a[2];
    }
    
    for (vector<vector<int> >::reverse_iterator it=network.rbegin();it!=network.rend();it++)
    {
        a=*it;
        if (head[a[1]]==maxn) head[a[1]]=a[2];
        sum[a[0]]+=head[a[1]];
        
        if ((it+1)!=network.rend())
        {
            b=*(it+1);
            if (b[0]==a[0]) continue;
        }

        head[a[0]]=min(head[a[0]],sum[a[0]]);
    }
  
    return head[sourceNode];
}