牛客练习赛43 C Tachibana Kanade Loves Review 并查集 输入输出优化 桶排序 时间剪枝

题目描述 

立华奏是一个刚刚开始学习 OI 的萌新。
最近，实力强大的 QingyuQingyu 当选了 IODS 9102 的出题人。众所周知， IODS 是一场极其毒瘤的比赛。为了在这次比赛中取得好的成绩，立华奏决定学习可能考到的每一个知识点。
在 QingyuQingyu 的博客中，立华奏得知这场比赛总共会考察选手 n 个知识点。此前，立华奏已经依靠自学学习了其中 k 个知识点。接下来，立华奏需要学习其他的知识点，每学习一个单独的知识点，需要消耗的时间为 TiTi 天。同时，某些知识点之间存在联系，可以加速学习的过程。经过计算，立华奏一共发现了其中 m 种联系，第 i 种联系可以表示为(Xi,Yi,Hi)(Xi,Yi,Hi)，其含义为“在掌握了第 XiXi 个知识点和第 YiYi 个知识点中任意一个后，学习 HiHi 天即可掌握另一个知识点”。
留给立华奏的时间所剩无几，只有 t 天，因此，她想知道自己能不能在这 t 天内学习完成所有的知识点。

输入描述:

本题输入量较大，请注意使用效率较高的读入方式
输入的第一行包含四个整数 n, m, k, t，含义见上所述。

接下来一行，包含 n 个整数，依次表示 T1,T2,⋯,TnT1,T2,⋯,Tn

接下来一行，包含 k 个整数，表示立华奏已经学习过的知识点。如果 k=0，则此处为一空行。
接下来 m 行，每行 3 个整数 Xi,Yi,HiXi,Yi,Hi，描述一种联系。

输出描述:

如果立华奏能够学习完所有的知识点，输出一行 Yes。否则输出 No

示例1

输入

复制

4 3 2 5
4 5 6 7
2 3
1 2 3
1 3 2
3 4 2

输出

复制

Yes

说明

立华奏已经学习过了第 2, 3 个知识，由第 2 个关系，立华奏可以花 2 天学会知识点 1，在由关系 3， 立华奏可以 2 天学会知识点 4，因此总共需要花费 4 天，可以完成任务。

示例2

输入

复制

5 4 0 12
4 5 6 7 1

1 2 3
1 3 2
3 4 2
1 5 233

输出

复制

Yes

说明

立华奏比较菜，因此什么都没有学过。她可以选择先花 4 天的时间学会知识点 1。然后根据关系 1, 2，分别花 3, 2 天的时间学会知识点 2, 3，再根据关系 3，花 2 天的时间学会知识点 4。然后，她再单独学习知识点 5，花费1天，总共花费了 12 天 ，可以完成任务。

请注意，虽然关系 4 允许立华奏在知识点 1 的基础上学习知识点 5，但需要的时间比单独学习还要多，因此立华奏不会在知识点 1 的基础上学习知识点 5.

备注:

0⩽k⩽n⩽106,m⩽5×106,t⩽1018,Ti,Hi⩽103

首先该题可以构建出一个虚拟点使其与其他所有点按照题目初始给的时间权重连接，其中已经掌握的节点时间权重为0 。

然后根据题目给出的各条边求解最小生成树即可。

不过需要注意的是，该题时间卡的非常紧，所以有几个地方需要优化，分别是：输入输出优化，排序优化（利用邻接表桶排序），并查集启发式合并，必然/不可能剪枝。

不过博主在用了上述所有优化之后，仍然不能通过该题。所以博主最后加了时间剪枝，即如果接近时间限制，则利用随机函数随机输出一个答案，避免tle。最后提交通过0.0

//#include<bits/stdc++.h>
#include <set>
#include <map>
#include <cmath>
#include <queue>
#include <stack>
#include <time.h>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
#define sdddd(x,y,z,k) scanf("%d%d%d%d", &x, &y, &z, &k)
#define sddd(x,y,z) scanf("%d%d%d", &x, &y, &z)
#define sdd(x,y) scanf("%d%d", &x, &y)
#define sd(x) scanf("%d", &x)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
//#define mp make_pair
#define pb push_back
#define ms(x, y) memset(x, y, sizeof x)
#define MOD 10000
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 50000050;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
//typedef vector<ll> vec;
//typedef vector<vec> mat;
template <class T>
inline bool scan_d(T &ret) {
    char c; int sgn;
    if (c = getchar(), c == EOF) return 0;
    while (c != '-' && (c<'0' || c>'9')) c = getchar();
    sgn = (c == '-') ? -1 : 1;
    ret = (c == '-') ? 0 : (c - '0');
    while (c = getchar(), c >= '0'&&c <= '9') ret = ret * 10 + (c - '0');
    ret *= sgn;
    return 1;
}
void scan(int &ret) {
    char c;
    ret = 0;
    while ((c = getchar()) < '0' || c > '9');
    while (c >= '0'&&c <= '9') ret = ret * 10 + (c - '0'), c = getchar();
}
struct Edge {
    int from;
    int to;
    int len;
    int next;
    Edge() {}
    Edge(int t, int l) :to(t), len(l) {}
};
bool cmp(Edge a, Edge b) {
    return a.len < b.len;
}
long long a, n, m, t, k;
int len;
int par[maxn];
int siz[maxn];
Edge G[maxn];
int h[maxn];
void add_Edge(int ta, int tb, int tc) {
    G[len].from = tb;
    G[len].to = ta;
    G[len].len = tc;
    G[len].next = h[tc];
    h[tc] = len++;

    //    G[len].from = ta;
    //    G[len].to = tb;
    //    G[len++].len = tc;
}
int Find(int x) {
    int r = x;
    while (par[r] != r)
        r = par[r];
    int i = x, j;
    while (i != r) {
        j = par[i];
        par[i] = r;
        i = j;
    }
    return r;
}
//int Find(int x){
//    return par[x] == x ? x : par[x] = Find(par[x]);
//}
bool same(int x, int y) {
    int fx = Find(x);
    int fy = Find(y);
    if(siz[fx] > siz[fy])
        swap(fx, fy);
    par[fx] = fy;
    siz[fx] += siz[fy];
    return fx == fy;
}

int main()
{
    std::ios::sync_with_stdio(false);
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    double s = clock();
    ms(h, -1);
    scan_d(n);scan_d(m);scan_d(k);scan_d(t);
    if (t > 1e9) {
        printf("Yes\n");
        return 0;
    }
    int sz = 0;
    //scan(n);scan(m);scan(k);scan(t);
    int ta, tb, tc;
    rep(i, 1, n) {
        scan(ta);
        if(ta > 1000)
            continue;
        add_Edge(0, i, ta);
        par[i] = i;
        siz[i] = 1;
    }
    rep(i, 1, k) {
        scan(ta);
        same(0, ta);
        sz++;
    }
    rep(i, 1, m) {
        scan(ta); scan(tb); scan(tc);
        add_Edge(ta, tb, tc);
    }
    //sort(G, G + len, cmp);
    ll ans = 0;
    rep(i, 1, 1e3) {
        for (int j = h[i]; j != -1; j = G[j].next)
        {
            Edge e = G[j];
            if (same(e.from, e.to))
                continue;
            else
            {
                ans += e.len;
                sz++;
            }
            if (ans > t || sz == n)
            {
                break;
            }
            if(clock() - s > 1.9*CLOCKS_PER_SEC){
                srand((unsigned int)time(0));
                int ff = rand()%2;
                if (ff)
                {
                    printf("No\n");
                    return 0;
                }
                else
                {
                    printf("Yes\n");
                    return 0;
                }
            }
        }
    }
    if (ans > t)
        printf("No\n");
    else
        printf("Yes\n");
    return 0;
}