HDU多校day3-1010Road Discount

Road Discount

题意：

$n$个点$m$条边，每条边有一个原价$a$和折扣$b$。你最多只能使用$k$ 次折扣，对$k\in [0,n)$输出最少的花费使图连通。

思路：

用最小的花费使图连通就是MST，然后原价边和折扣边就可以看成黑白边。问题就是恰好包含$k$条白边的最小生成树。
那么对于每条白边，我们通过对其权值加上一个$c$。由于边权的范围是$[1,1000]$，那么对于每个$c\in [-1000,1000]$，都做一次最小生成树。在排序的时候，如果权值相同，则优先选择黑边，因为白边可以替换黑边，所以我们统计最少需要多少白边。
前面可以做两次最小生成树，因为非树边在新的最小生成树中也一定不会用到，所以我们就可以将边的数量变成$2n-2$。
最后，通过枚举$k$,就可以算出最小的代价。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define ll long long
#define mp make_pair
#define pb push_back
#define ls x<<1
#define rs x<<1|1
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define pii pair<int,int>
#define all(x) x.begin(),x.end()
#define cl(x,y) memset(x,y,sizeof(x))
#define nxtp(a,n) next_permutation(a+1,a+n+1)
#define mem(x,y,n) memset(x,y,sizeof(int)*(n+5))
const int N=1e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
const double INF=1e18;
using namespace std;

struct edge
{
    int u,v,a,b,w,w2,col;
}e[N],ed[N];
int fa[N],n,m,cnt;
int ans[N][2];
int find(int x)
{
    if(fa[x]!=x)
        fa[x]=find(fa[x]);
    return fa[x];
}
void reduce(int col)
{
    int i,tot=0;
    for(i=1;i<=n;i++)
        fa[i]=i;
    for(i=1;i<=m;i++)
    {
        int t1=find(ed[i].u),t2=find(ed[i].v);
        if(t1==t2)
            continue;
        fa[t1]=t2;
        e[++cnt]=ed[i];
        if(col==0)
            e[cnt].w=e[cnt].w2=ed[i].a;
        else    
            e[cnt].w=e[cnt].w2=ed[i].b;
        e[cnt].col=col;
        tot++;
        if(tot==n-1)
            break;
    }
    return;
}
int ask(int k)
{
    int i;
    for(i=-1000;i<=1000;i++)
        if(ans[i+1000][1]<=k)
             return ans[i+1000][0]-k*i;                  
    return -1;
}
void add(int x)
{
    int i;
    for(i=1;i<=n;i++)
        fa[i]=i;
    for(i=1;i<=cnt;i++)
        if(e[i].col==1)
            e[i].w2=e[i].w+x;
    sort(e+1,e+cnt+1,[](edge x,edge y){
        if(x.w2==y.w2)
            return x.col<y.col;
        return x.w2<y.w2;
    });
    int sum=0,tot=0,one=0;
    for(i=1;i<=cnt;i++)
    {
        int t1=find(e[i].u),t2=find(e[i].v);
        if(t1==t2)
            continue;
        fa[t1]=t2;
        sum+=e[i].w2;
        if(e[i].col==1)
            one++;
        tot++;
        if(tot==n-1)
            break;
    }
    ans[x+1000][0]=sum;
    ans[x+1000][1]=one;
}
int main()
{
#ifdef ChuTian
    freopen("in.in","r",stdin);
    freopen("out.out","w",stdout);
#endif
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int T;
	cin>>T;
	while(T--)
	{
        int i;
        cin>>n>>m;
        for(i=1;i<=m;i++)
            cin>>ed[i].u>>ed[i].v>>ed[i].a>>ed[i].b;
        cnt=0;
        sort(ed+1,ed+m+1,[](edge x,edge y){
            return x.a<y.a;
        });
        reduce(0);
        sort(ed+1,ed+m+1,[](edge x,edge y){
            return x.b<y.b;
        });
        reduce(1);  
        for(i=-1000;i<=1000;i++)
            add(i);
        for(i=0;i<n;i++)
            cout<<ask(i)<<endl;
	}
	return 0;
}