注意到只有q更大的人才能成爲q較小的人的父節點,所以整個樹的根節點一定是q最大的那個人。
把所有人按q從大到小排序,每次加入到圖中q最大的人,因爲所有q比他大的人都已經加入到圖中,所以所有有可能成爲其父節點的人都在圖中了。反言之,如果圖中存在可能成爲其父節點的人,則計算其中cost最小的那個加入到ans中,反之如果不存在,則無解,ans=-1。
#include <cstdio>
#include <cstring>
#include <cctype>
#include <cstdlib>
#include <ctime>
#include <climits>
#include <cmath>
#include <iostream>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <list>
#include <queue>
#include <stack>
#include <deque>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1010;
struct Node
{
int q,id;
bool operator<(const Node &o) const
{
return q>o.q;
}
};
struct Edge
{
int v,d;
Edge(int v,int d):v(v),d(d){}
};
int n,m;
Node a[maxn];
vector<Edge> g[maxn];
bool vis[maxn];
int ans;
int main()
{
while (scanf("%d",&n)==1)
{
for (int i=1;i<=n;i++)
{
scanf("%d",&a[i].q);
a[i].id=i;
}
scanf("%d",&m);
while (m--)
{
int u,v,c;
scanf("%d%d%d",&u,&v,&c);
g[v].push_back(Edge(u,c));
}
sort(a+1,a+n+1);
memset(vis,0,sizeof(vis));
ans=0;
vis[a[1].id]=1;
for (int i=2;i<=n;i++)
{
int u=a[i].id, mm=INF;
for (int j=0;j<g[u].size();j++)
{
int v=g[u][j].v, d=g[u][j].d;
if (vis[v]) mm=min(mm,d);
}
if (mm==INF) {ans=-1; break;}
ans+=mm;
vis[u]=1;
}
printf("%d\n",ans);
}
return 0;
}