bzoj3454

http://www.elijahqi.win/archives/696
Description

阿狸和桃子养了n个小阿狸, 小阿狸们每天都在一起玩的很开心. 作为工程师的阿狸在对小阿狸们之间的关系进行研究以后发现了小阿狸的人际关系由某种神奇的相互作用决定, 阿狸称之为“键”. 每个键有一个频率, 称为键频率, 是一个整数(单位Hz).
由于小阿狸们每天成集团地黏在一起, 桃子希望他们能够分成更加独立的几团. 阿狸发现, 一旦小阿狸们分开, 独立的一块连在一起的几个小阿狸就会形成一个家族, 而家族的类型由这个家族的小阿狸的数量唯一确定(比如说只有一个小阿狸的家族显然就是单身码农, 两个小阿狸的显然是一对小阿狸恋人, 三个小阿狸的就是三口之家等等). 显然, 一个小阿狸和另一个小阿狸处于同一家族, 当且仅当两个小阿狸之间存在直接或间接的键组成的路径.
桃子对每种小阿狸家族都有自己的喜好程度, 她希望所有的小阿狸家族喜好程度之和大于等于K.
为了让小阿狸们分开来, 阿狸决定让某些键断裂, 只保留某一段频率的键, 比如说100Hz到140Hz频率的键, 这时频段宽度为40Hz. 当然, 阿狸希望频段宽度越小越好, 但至少要有一个小键. 你的任务就是求出最小的频段宽度.
注意, 输入不保证全部键都有效时只有一个小阿狸家族.

Input

第一行3个整数n(<=1000), m(<=5000), K(0~2^31-1).
接下来1行n个整数, 第k的整数表示桃子对大小为k的小阿狸家族的喜爱程度.
接下来m行, 每行3个整数, u, v, f. 表示u小阿狸和小阿狸v之间存键, 频率f Hz.

Output

一个整数, 即最窄的频段宽度(不存在可行频段, 输出”T_T”, 不含引号).

Sample Input

4 4 52
1 50 2 9
1 2 6
2 3 8
3 4 4
1 4 3
Sample Output

0
(解释:频段3Hz~3Hz或4Hz~4Hz或6Hz~6Hz或8Hz~8Hz)

HINT

对于 100% 的数据, n <=1000 , m <=3000

首先我们考虑暴力枚举去做，那么应该是m！的复杂度 就是我们每次去枚举删几条边，删的都是哪些边，然后再统计但显然速度太慢。

我们考虑将所有的键的频率按照从小到大的顺序去排序 首先我们得先记录一个当所有人都是单身的时候这个满意度

然后开始从小到大加入键，我们用并查集去维护每个区间的连通性，并且设立size数组表示这个连通块的点的个数

要连接的时候把原来的好感度减去加上新组合的家族的好感度

由于我们固定左端点，当右端点不断递增并且满足的时候我们后面就可以退出不做了，因为这样可以保障我的f的差是最小的

#include<cstdio>
#include<algorithm>
#define inf 0x7fffffff
#define M 5500
#define N 1100
inline int read(){
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
struct node{
    int x,y,f;
}data[M];
int fa[N],size[N],k[N];
inline int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
inline bool cmp(node a,node b){return a.f<b.f;}
int n,m,kk,ans,sum;
int main(){
    freopen("bzoj3454.in","r",stdin);
    n=read();m=read();kk=read();
    for (int i=1;i<=n;++i) k[i]=read();
    for (int i=1;i<=m;++i) data[i].x=read(),data[i].y=read(),data[i].f=read();
    std::sort(data+1,data+m+1,cmp); 
    ans=inf;
    for (int i=1;i<=m;++i){
        if (data[i].f==data[i-1].f) continue;
        for (int j=1;j<=n;++j) fa[j]=j,size[j]=1;sum=n*k[1];//size用来统计每个连通块中的个数 
        for (int j=i;j<=m;++j){
            if (data[j].f-data[i].f>=ans) break;
            int xx=find(data[j].x),yy=find(data[j].y);
            if (xx==yy) continue;
            sum-=k[size[xx]]+k[size[yy]];sum+=k[size[xx]+size[yy]];
            fa[xx]=yy;size[yy]+=size[xx];
            if(sum>=kk) {ans=data[j].f-data[i].f;break;}
        }
    }
    if (ans!=inf) printf("%d",ans);else printf("T_T"); 
    return 0;
}