HDU-2448 Mining Station on the Sea

#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <fstream>
#include <iostream>
#include <vector>
#include <cctype>
#include <queue>
#define rep(i, l, r) for(int i=l; i<=r; i++)
#define clr(x, c) memset(x, c, sizeof(x))
#define N 234
#define MAX 0x3fffffff
#define ll long long
using namespace std;
int read()
{
	int x=0, f=1; char ch=getchar();
	while (!isdigit(ch)) { if (ch==‘-‘) f=-1; ch=getchar(); }
	while (isdigit(ch)) { x=x*10+ch-‘0‘; ch=getchar(); }
	return x*f;
}

int n, m, en, p, l[N], st[N], lx[N], ly[N], v[N][N], k[N], d[N][N], w[N];
bool vx[N], vy[N];

bool Find(int x)
{
	vx[x]=1;
	rep(y, 1, n) 
	{
		if (vy[y]) continue;
		int a=lx[x]+ly[y]-v[x][y];
		if (!a)
		{
			vy[y]=1; if (!l[y] || Find(l[y])) { l[y]=x; return 1; }
		}
		else st[y]=min(st[y], a);
	}
	return false;
}

inline int km()
{
	clr(l, 0); rep(i, 1, n) lx[i]=-MAX;
	rep(i, 1, n) rep(j, 1, n) if (lx[i]<v[i][j]) lx[i]=v[i][j]; rep(i, 1, n) if (lx[i]==-MAX) return 1;
	clr(ly, 1); rep(i, 1, n) rep(j, 1, n) if (v[i][j]!=-MAX) ly[j]=0; rep(i, 1, n) if (ly[i]) return 1;
	rep(i, 1, n) 
	{
		rep(j, 1, n) st[j]=MAX;
		while (1)
		{
			clr(vx, 0); clr(vy, 0);
			if (Find(i)) break; int a=MAX;
			rep(j, 1, n) if (!vy[j] && st[j]<a) a=st[j]; 
			if (a==MAX) return 1;
			rep(j, 1, n) if (vx[j]) lx[j]-=a;
			rep(j, 1, n) if (vy[j]) ly[j]+=a; else st[j]-=a;
		}
	}
	int a=0; rep(i, 1, n) a+=lx[i]+ly[i];
	return a;
}

int main()
{
	while (~scanf("%d%d%d%d", &n, &m ,&en, &p))
	{
		rep(i, 1, m) rep(j, 1, m) d[i][j] = i==j?0:MAX;
		rep(i, 1, n) w[i]=read();
		rep(i, 1, en) { int x=read(), y=read(), z=read(); if (d[x][y]>z) d[x][y]=d[y][x]=z; }
		rep(o, 1, m) rep(i, 1, m) rep(j, 1, m) if (d[i][o]+d[o][j]<d[i][j]) d[i][j]=d[i][o]+d[o][j];
		
		rep(i, 1, n) rep(j, 1, n) v[i][j]=MAX;
		rep(j, 1, p) 
		{
			int x=read(), y=read(), z=read();
			rep(i, 1, n) if (v[i][x]>d[w[i]][y]+z) 
				v[i][x]=d[w[i]][y]+z;
		}
		rep(i, 1, n) rep(j, 1, n) v[i][j]*=-1;
		printf("%d\n", -km());
	}
	return 0;
}