标签：图论 int YbtOJ ans 最小 vis cnt head qwq
为什么区间 dp 又咕咕咕了QAQ
于是随机抽取了一个幸运章节来做。
目前处于半摆烂状态。
例题1.繁忙都市
板子。写了下以前几乎没写过的堆优化 Prim。
code
#include<bits/stdc++.h>
#define pii pair<int,int>
#define fi first
#define se second
using namespace std;
const int N=300;
const int M=2e5+5;
int n,m;
int head[N],cnt;
struct node{
int nxt,to,w;
}e[M];
void add(int u,int v,int w){
e[++cnt].nxt=head[u];e[cnt].to=v;e[cnt].w=w;head[u]=cnt;
}
int vis[N];
priority_queue<pii,vector<pii>,greater<pii> >q;
int Prim(int s)
{
int ans=0;
q.push(pii(0,s));
while(!q.empty())
{
int w=q.top().fi,u=q.top().se;q.pop();
if(vis[u]) continue;
ans=max(ans,w);vis[u]=1;
//cout<<w<<" "<<u<<endl;
for(int i=head[u];i;i=e[i].nxt)
{
int v=e[i].to;
if(vis[v]) continue;
q.push(pii(e[i].w,v));
}
}
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1,u,v,w;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);add(v,u,w);
}
cout<<n-1<<" "<<Prim(1)<<endl;
return 0;
}
例题2.新的开始
每个点往 0 号点连一条边代表建立电站，跑最小生成树。
code
#include<bits/stdc++.h>
#define pii pair<int,int>
#define fi first
#define se second
using namespace std;
const int N=305;
const int M=1e5+5;
int n,m,a[N];
int head[N],cnt;
struct node{
int nxt,to,w;
}e[M];
void add(int u,int v,int w){
e[++cnt].nxt=head[u];e[cnt].to=v;e[cnt].w=w;head[u]=cnt;
}
int vis[N],tot;
priority_queue<pii,vector<pii>,greater<pii> >q;
int Prim(int s)
{
int ans=0;
q.push(pii(0,s));
while(!q.empty())
{
int w=q.top().fi,u=q.top().se;q.pop();
if(vis[u]) continue;
ans+=w;vis[u]=1;tot++;
for(int i=head[u];i;i=e[i].nxt)
{
int v=e[i].to;
if(vis[v]) continue;
q.push(pii(e[i].w,v));
}
}
return ans;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]),add(i,0,a[i]),add(0,i,a[i]);
for(int i=1;i<=n;i++)
{
for(int j=1,w;j<=n;j++)
{
scanf("%d",&w);
if(i!=j) add(i,j,w);
}
}
cout<<Prim(0)<<endl;
//cout<<n-1<<" "<<Prim(1)<<endl;
return 0;
}
例题3.公路建设
原先就不在最小生成树上的边，加了其他边也还是不在。
只考虑最小生成树原有的边和新加进来这条就好了。
code
#include<bits/stdc++.h>
using namespace std;
const int N=2005;
int n,m,fa[N];
struct node{
int u,v,w;
}e[N];
int find(int x)
{
if(fa[x]==x) return x;
else return fa[x]=find(fa[x]);
}
int cnt,ans;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=1;i<=m;i++)
{
cnt++;
scanf("%d%d%d",&e[cnt].u,&e[cnt].v,&e[cnt].w);
if(find(e[cnt].u)!=find(e[cnt].v))
{
fa[find(e[cnt].u)]=find(e[cnt].v);
ans+=e[cnt].w;
if(cnt==n-1) printf("%.1lf\n",ans/2.0);
else cout<<0<<endl;
int qwq=cnt;
while(e[qwq].w<e[qwq-1].w&&qwq>1)
{
swap(e[qwq],e[qwq-1]);
qwq--;
}
continue;
}
int qwq=cnt;
while(e[qwq].w<e[qwq-1].w&&qwq>1)
{
swap(e[qwq],e[qwq-1]);
qwq--;
}
//for(int i=1;i<=cnt;i++) cout<<e[i].u<<" "<<e[i].v<<" "<<e[i].w<<endl;
ans=0;int flag=0,now=cnt;cnt=0;
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=1;i<=now;i++)
{
int u=e[i].u,v=e[i].v,w=e[i].w;
if(find(u)==find(v)) {flag=i;continue;}
fa[find(u)]=find(v);cnt++;
ans+=w;
}
while(flag<=cnt)
{
swap(e[flag],e[flag+1]);
flag++;
}
if(cnt==n-1) printf("%.1lf\n",ans/2.0);
else cout<<0<<endl;
}
return 0;
}
标签：图论,int,YbtOJ,ans,最小,vis,cnt,head,qwq 来源： https://www.cnblogs.com/ying-xue/p/16596738.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。