社区讨论

求助

P14362[CSP-S 2025] 道路修复参与者 2已保存回复 1

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
1 条
当前快照
1 份
快照标识符
@mhiyda5q
此快照首次捕获于
2025/11/03 17:43
4 个月前
此快照最后确认于
2025/11/03 17:43
4 个月前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
#define ll long long
struct Edge
{
    int u,v,e;
    ll w;
}edge[110010],edg[1000010];
int n,m,k;
ll c[15],num,ans=0;
int fa[110010];
int find(int x){return fa[x]==x?x:find(fa[x]);}
void merge(int x,int y){fa[find(x)]=find(y);}
bool cmp(Edge x,Edge y){return x.w<y.w;}
int main()
{
    std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>m>>k;
    for(int i=1;i<=m;++i)cin>>edg[i].u>>edg[i].v>>edg[i].w;
    sort(edg+1,edg+1+m,cmp);
    for(int i=1;i<=n;++i)fa[i]=i;
    for(int i=1,j=0;i<=m;++i)
    {
        if(find(edg[i].u)==find(edg[i].v))continue;
        edge[++j]=edg[i];
        edge[j].e=0;
        merge(edg[i].u,edg[i].v);
        ans+=edg[i].w;
    }
    for(int i=1,j=n;i<=k;++i)
    {
        cin>>c[i];
        for(int k=1;k<=n;++k,++j)
        {
            cin>>edge[j].w;
            edge[j].u=n+i;
            edge[j].v=k;
            edge[j].e=i;
        }
    }
    num=n+n*k-1;
    sort(edge+1,edge+1+num,cmp);
    // for(int i=1;i<=num;++i)cout<<edge[i].u<<" "<<edge[i].v<<" "<<edge[i].w<<" "<<edge[i].e<<"\n";
    for(int s=1;s<(1<<k);++s)
    {
        for(int i=1;i<=n+k;++i)fa[i]=i;
        ll res=0,cnt=0;
        for(int i=1;i<=k;++i)
        {
            res+=c[i]*(s&(1<<(i-1)));
            cnt-=(s&(1<<(i-1)));
        }
        for(int i=1;i<=num;++i)
        {
            if(edge[i].e!=0&&(s&(1<<(edge[i].e-1)))==0)continue;
            if(find(edge[i].u)==find(edge[i].v))continue;
            res+=edge[i].w;
            merge(edge[i].u,edge[i].v);
            ++cnt;
            if(res>ans||cnt==n)break;
        }
        ans=min(ans,res);
    }
    cout<<ans;
    return 0;
}
预估复杂度O(2^k*nklog(nk))为什么k=0还会TLE

回复

1 条回复,欢迎继续交流。

正在加载回复...