图论12-无向带权图及实现

文章目录

带权图

1.1带权图的实现

在无向无权图的基础上,增加边的权。

使用TreeMap存储边的权重。

  • 遍历输入文件,创建TreeMap adj存储每个节点。
  • 每个输入的adj节点链接新的TreeMap,存储相邻的边和权重
c 复制代码
private TreeMap<Integer, Integer>[] adj;

adj = new TreeMap[V];

for(int i = 0; i < V; i ++)
    adj[i] = new TreeMap<Integer, Integer>();
  • 两条边相连,则分别把权重加入各自的邻接表中
c 复制代码
adj[a].put(b, weight);
adj[b].put(a, weight);
  • 判断两点之间是否有边
c 复制代码
public boolean hasEdge(int v, int w){
    validateVertex(v);
    validateVertex(w);
    return adj[v].containsKey(w);
}
  • 求相邻的所有节点
c 复制代码
public Iterable<Integer> adj(int v){
    validateVertex(v);
    return adj[v].keySet();
}
  • 求两点的权值
c 复制代码
public int getWeight(int v, int w){

    if(hasEdge(v, w)) return adj[v].get(w);
    throw new IllegalArgumentException(String.format("No edge %d-%d", v, w));
}
  • 移除边
c 复制代码
public void removeEdge(int v, int w){
    validateVertex(v);
    validateVertex(w);
    if(adj[v].containsKey(w)) E --;
    adj[v].remove(w);
    adj[w].remove(v);
}
  • 复制一个图
c 复制代码
public Object clone(){

    try{
        WeightedGraph cloned = (WeightedGraph) super.clone();
        cloned.adj = new TreeMap[V];
        for(int v = 0; v < V; v ++){
            cloned.adj[v] = new TreeMap<Integer, Integer>();
            for(Map.Entry<Integer, Integer> entry: adj[v].entrySet())
                cloned.adj[v].put(entry.getKey(), entry.getValue());
        }
        return cloned;
    }
    catch (CloneNotSupportedException e){
        e.printStackTrace();
    }
    return null;
}

1.2 完整代码

c 复制代码
package Chapter09_Weight_Graph;

import java.io.File;
import java.io.IOException;
import java.util.Map;
import java.util.TreeMap;
import java.util.Scanner;


/// 暂时只支持无向带权图
public class WeightedGraph implements Cloneable{

    private int V;
    private int E;
    private TreeMap<Integer, Integer>[] adj;

    public WeightedGraph(String filename){

        File file = new File(filename);

        try(Scanner scanner = new Scanner(file)){

            V = scanner.nextInt();
            if(V < 0) throw new IllegalArgumentException("V must be non-negative");
            adj = new TreeMap[V];
            for(int i = 0; i < V; i ++)
                adj[i] = new TreeMap<Integer, Integer>();

            E = scanner.nextInt();
            if(E < 0) throw new IllegalArgumentException("E must be non-negative");

            for(int i = 0; i < E; i ++){
                int a = scanner.nextInt();
                validateVertex(a);
                int b = scanner.nextInt();
                validateVertex(b);
                int weight = scanner.nextInt();

                if(a == b) throw new IllegalArgumentException("Self Loop is Detected!");
                if(adj[a].containsKey(b)) throw new IllegalArgumentException("Parallel Edges are Detected!");

                adj[a].put(b, weight);
                adj[b].put(a, weight);
            }
        }
        catch(IOException e){
            e.printStackTrace();
        }
    }

    public void validateVertex(int v){
        if(v < 0 || v >= V)
            throw new IllegalArgumentException("vertex " + v + "is invalid");
    }

    public int V(){
        return V;
    }

    public int E(){
        return E;
    }

    public boolean hasEdge(int v, int w){
        validateVertex(v);
        validateVertex(w);
        return adj[v].containsKey(w);
    }

    public Iterable<Integer> adj(int v){
        validateVertex(v);
        return adj[v].keySet();
    }

    public int getWeight(int v, int w){

        if(hasEdge(v, w)) return adj[v].get(w);
        throw new IllegalArgumentException(String.format("No edge %d-%d", v, w));
    }

    public int degree(int v){
        validateVertex(v);
        return adj[v].size();
    }

    public void removeEdge(int v, int w){
        validateVertex(v);
        validateVertex(w);
        if(adj[v].containsKey(w)) E --;
        adj[v].remove(w);
        adj[w].remove(v);
    }

    @Override
    public Object clone(){

        try{
            WeightedGraph cloned = (WeightedGraph) super.clone();
            cloned.adj = new TreeMap[V];
            for(int v = 0; v < V; v ++){
                cloned.adj[v] = new TreeMap<Integer, Integer>();
                for(Map.Entry<Integer, Integer> entry: adj[v].entrySet())
                    cloned.adj[v].put(entry.getKey(), entry.getValue());
            }
            return cloned;
        }
        catch (CloneNotSupportedException e){
            e.printStackTrace();
        }
        return null;
    }

    @Override
    public String toString(){
        StringBuilder sb = new StringBuilder();

        sb.append(String.format("V = %d, E = %d\n", V, E));
        for(int v = 0; v < V; v ++){
            sb.append(String.format("%d : ", v));
            for(Map.Entry<Integer, Integer> entry: adj[v].entrySet())
                sb.append(String.format("(%d: %d) ", entry.getKey(), entry.getValue()));
            sb.append('\n');
        }
        return sb.toString();
    }

    public static void main(String[] args){

        WeightedGraph g = new WeightedGraph("gw1.txt");
        System.out.print(g);
    }
}
相关推荐
是糖不是唐19 小时前
代码随想录算法训练营第五十三天|Day53 图论
c语言·数据结构·算法·图论
vir022 天前
好奇怪的游戏(BFS)
数据结构·c++·算法·游戏·深度优先·图论·宽度优先
一个不喜欢and不会代码的码农2 天前
李春葆《数据结构》——图相关代码
数据结构·算法·图论
是糖不是唐2 天前
代码随想录算法训练营第五十二天|Day52 图论
c语言·算法·深度优先·动态规划·图论
南宫生2 天前
力扣-Hot100-图论【算法学习day.38】
java·学习·算法·leetcode·链表·图论
张焚雪2 天前
关于图论建模的一份介绍
python·数学建模·图论
是糖不是唐3 天前
代码随想录算法训练营第五十一天|Day51 图论
c语言·数据结构·算法·深度优先·图论
汉克老师4 天前
GESP4级考试语法知识(贪心算法(六))
开发语言·数据结构·c++·算法·贪心算法·图论
是糖不是唐4 天前
代码随想录算法训练营第五十天|Day50 图论
c语言·数据结构·算法·图论
Romanticroom4 天前
图论之最小生成树计数(最小生成树的应用)
算法·图论