综合（奇技淫巧）：HDU 5118 GRE Words Once More!

GRE Words Once More!

Time Limit: 5000/5000 MS (Java/Others)    Memory Limit: 512000/512000 K (Java/Others)
Total Submission(s): 205    Accepted Submission(s): 32

Problem Description

Now Matt is preparing for the Graduate Record Examinations as Coach Pang did in 2013 and George did in 2011.

Thanks to modern techniques, Matt uses automata instead of old-fasioned vocabulary books.

The automata used by Matt is a directed acyclic graph (DAG) with N vertices and M edges. The vertices are conveniently numbered by 1, 2, . . . , N . Each edge is labeled with an integer. Additionally, some vertices are marked as special.

A GRE word is obtained by concatenating the labels on the path from vertex 1 to a special vertex.

Now, Matt has Q questions. The i-th question is asking for the length of ki-th smallest words among all the GRE words he can obtain in lexicographical order.

 

Input

The first line contains only one integer T , which indicates the number of test cases.

For each test case, the first line contains three integers N, M, Q (2 ≤ N ≤ 10⁵, 0 ≤ M ≤ 10⁵, 1 ≤ Q ≤ 10⁵).

The second line contains N - 1 integers s₂, . . . , s_n. If the i-th vertex is special, then s_i = 1. Otherwise, s_i = 0. Vertex 1 is never special.

Each of the following M lines contains three integers a_i, b_i, c_i denoting an edge from vertex ai to vertex b_i labeled with c_i (1 ≤ a_i, b_i ≤ N, 1 ≤ c_i ≤ 10⁹). For each vertex v, all outgoing edges are labeled with distinct integers.

Each of the following Q lines contains the integer ki (1 ≤ k_i ≤ 10⁸) of the i-th question.

 

Output

For each test case, output “Case #x:” in the frirst line, where x is the case number (starting from 1).

Then, for each question, output the length of the word in one line. If the word does not exist, output “-1” (without quotes) instead.

 

Sample Input

1 3 3 4 1 1 1 2 1 1 3 12 2 3 3 1 2 3 4

 

Sample Output

Case #1: 1 2 1 -1

Hint

There are 3 GRE words in total (sorted in lexicographical order): 1. (1) 2. (1, 3) 3. (12)

　　这道题不是很难，需要注意清空数组。

　　思路是预处理答案，DFS时用手写栈防爆栈，有个必要的优化，就是扫过后答案是可以重复利用的。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
const int N=200010,M=100000000;
vector<pair<int,int> >g[N];
int ans[M+10],f[N],be[N],ed[N],tot;
int st[N],dep[N],vis[N],mem[N],top;
int T,cas=0,q,n,m,Q;
int main(){
    scanf("%d",&T);    
    while(T--){
        scanf("%d%d%d",&n,&m,&Q);tot=0;
        for(int i=2;i<=n;i++)scanf("%d",&f[i]);
        for(int i=1,a,b,v;i<=m;i++){
            scanf("%d%d%d",&a,&b,&v);
            g[a].push_back(make_pair(v,b));
        }
        for(int i=1;i<=n;i++)
            sort(g[i].begin(),g[i].end());
        st[top=1]=1;dep[top]=0;
        memset(vis,0,sizeof(vis));
        memset(be,0,sizeof(be));
        memset(ed,0,sizeof(ed));
        while(top){
            int x=st[top],d=dep[top];
            if(vis[top]){
                if(!ed[x])ed[x]=tot;
                vis[top]=0;top-=1;
                continue;
            }
            vis[top]=1;
            if(be[x]){
                int depth=-mem[x]+d;
                for(int i=be[x];i<=ed[x];i++){
                    ans[++tot]=ans[i]+depth;
                    if(tot>=M)break;
                }if(tot>=M)break;
                continue;    
            }
            be[x]=tot+1;mem[x]=d;
            if(f[x])ans[++tot]=d;
            if(tot>=M)break;
            for(int i=g[x].size()-1;~i;i--){
                st[++top]=g[x][i].second;
                dep[top]=d+1;
            }
        }
        printf("Case #%d:
",++cas);
        while(Q--){
            scanf("%d",&q);
            if(q>tot)printf("-1
");
            else printf("%d
",ans[q]);
        }
        for(int i=1;i<=n;i++)g[i].clear();
    }
    return 0;
}