• poj 3164(最小树形图)


    题目链接:http://poj.org/problem?id=3164

    思路:朱刘算法,模版题。

      1 #include<iostream>
      2 #include<cstdio>
      3 #include<cstring>
      4 #include<algorithm>
      5 #include<cmath>
      6 using namespace std;
      7 #define MAXN 111
      8 #define inf 1<<30
      9 
     10 struct Edge{
     11     int u,v;
     12     double w;
     13 }edge[MAXN*MAXN];
     14 
     15 struct Node {
     16     double x,y;
     17 } node[MAXN];
     18 
     19 int n,m;
     20 double In[MAXN];
     21 int pre[MAXN],visited[MAXN],ID[MAXN];
     22 //root表示根结点,n是顶点树,m是边数
     23 //最小树形图邻接表版本,三步走,找最小入弧,找有向环,缩环为点
     24 double Directed_MST(int root,int n,int m)
     25 {
     26     int u,v,i,ansi;
     27     double cnt=0;
     28     while(true) {
     29         //找最小入边
     30         for(i=0; i<n; i++)In[i]=inf;
     31         for(i=0; i<m; i++) {
     32             u=edge[i].u;
     33             v=edge[i].v;
     34             if(edge[i].w<In[v]&&u!=v) {
     35                 pre[v]=u;//u->v;
     36                 if(u==root)//记录是root从哪一条边到有效点的(这个点就是实际的起点)
     37                     ansi=i;
     38                 In[v]=edge[i].w;
     39             }
     40         }
     41         for(i=0; i<n; i++) {
     42             if(i==root)continue;
     43             if(In[i]==inf)return -1;//说明存在点没有入边
     44         }
     45         //找环
     46         int cntcode=0;
     47         memset(visited,-1,sizeof(visited));
     48         memset(ID,-1,sizeof(ID));
     49         In[root]=0;
     50         //标记每一个环
     51         for(i=0; i<n; i++) {
     52             cnt+=In[i];
     53             v=i;
     54             while(visited[v]!=i&&ID[v]==-1&&v!=root) {
     55                 visited[v]=i;
     56                 v=pre[v];
     57             }
     58             //说明此时找到一个环
     59             if(v!=root&&ID[v]==-1) {
     60                 //表示这是找到的第几个环,给找到的环的每个点标号
     61                 for(u=pre[v]; u!=v; u=pre[u]) {
     62                     ID[u]=cntcode;
     63                 }
     64                 ID[v]=cntcode++;
     65             }
     66         }
     67         if(cntcode==0)break;//说明不存在环
     68         for(i=0; i<n; i++) {
     69             if(ID[i]==-1)
     70                 ID[i]=cntcode++;
     71         }
     72         //缩点,重新标记
     73         for(i=0; i<m; i++) {
     74             int v=edge[i].v;
     75             edge[i].u=ID[edge[i].u];
     76             edge[i].v=ID[edge[i].v];
     77             //说明原先不在同一个环
     78             if(edge[i].u!=edge[i].v) {
     79                 edge[i].w-=In[v];
     80             }
     81         }
     82         n=cntcode;
     83         root=ID[root];
     84     }
     85     return cnt;
     86 }
     87 
     88 double Get_dist(int i,int j)
     89 {
     90     double d1=(node[i].x-node[j].x)*(node[i].x-node[j].x);
     91     double d2=(node[i].y-node[j].y)*(node[i].y-node[j].y);
     92     return sqrt(d1+d2);
     93 }
     94 
     95 int main()
     96 {
     97     double ans;
     98     while(~scanf("%d%d",&n,&m)){
     99         for(int i=0;i<n;i++)scanf("%lf%lf",&node[i].x,&node[i].y);
    100         for(int i=0;i<m;i++){
    101             scanf("%d%d",&edge[i].u,&edge[i].v);
    102             edge[i].u--;
    103             edge[i].v--;
    104             if(edge[i].u!=edge[i].v)edge[i].w=Get_dist(edge[i].u,edge[i].v);
    105             else edge[i].w=inf;//消除自环
    106         }
    107         ans=Directed_MST(0,n,m);
    108         if(ans==-1){
    109             puts("poor snoopy");
    110         }else
    111             printf("%.2f
    ",ans);
    112     }
    113     return 0;
    114 }
    View Code
  • 相关阅读:
    transformer代码笔记----transformer.py
    transformer代码笔记----decoder.py
    transformer代码笔记----attention.py
    transformer代码笔记----encoder.py
    transformer代码笔记----train.py
    Transformer《Attention Is All You Need》的理论理解
    MNIST手写识别Python(详细注释)
    mysql之重用查询
    mysql之sql查询结果集小数保留
    java之jdk的安装
  • 原文地址:https://www.cnblogs.com/wally/p/3279500.html
Copyright © 2020-2023  润新知