#543895 (Java21) No.1207 グラフX

提出ソース

結果

問題	No.1207 グラフX
ユーザー	CuriousFairy315
提出日時	2020-08-30 14:07:01
言語	Java21 (openjdk 21)
結果	WA
実行時間	-
コード長	10,952 bytes
コンパイル時間	4,295 ms
コンパイル使用メモリ	91,216 KB
実行使用メモリ	113,196 KB
最終ジャッジ日時	2024-04-27 07:18:31
合計ジャッジ時間	31,067 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	716 ms 109,560 KB
testcase_01	AC	806 ms 113,196 KB
testcase_02	AC	775 ms 109,144 KB
testcase_03	AC	814 ms 102,532 KB
testcase_04	WA	-
testcase_05	AC	669 ms 108,204 KB
testcase_06	AC	677 ms 108,164 KB
testcase_07	AC	621 ms 104,732 KB
testcase_08	WA	-
testcase_09	WA	-
testcase_10	AC	691 ms 106,520 KB
testcase_11	AC	689 ms 106,868 KB
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	WA	-
testcase_36	WA	-
testcase_37	WA	-
testcase_38	WA	-
testcase_39	WA	-
testcase_40	WA	-
testcase_41	WA	-
testcase_42	AC	47 ms 49,940 KB
testcase_43	AC	46 ms 50,312 KB
testcase_44	AC	46 ms 50,004 KB
testcase_45	AC	653 ms 99,464 KB
testcase_46	AC	693 ms 99,756 KB
testcase_47	AC	669 ms 99,736 KB
testcase_48	AC	653 ms 99,564 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code


import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.BitSet;
import java.util.NoSuchElementException;

public class Main {

	public static void main(String[] args) {
		new Main();
	}

	public Main() {
		InputChecker ic = new InputChecker(System.in);
		java.io.PrintWriter out = new java.io.PrintWriter(System.out);
		solve(ic, out);
		out.flush();
	}

	public void solve(InputChecker ic, java.io.PrintWriter out) {
		int N = ic.nextInt(2, exponent10(2, 5));
		ic.nextChar(' ');
		int M = ic.nextInt(N - 1, exponent10(2, 5));
		ic.nextChar(' ');
		int X = ic.nextInt(2, exponent10(1, 9));
		ic.readNewLine();
		UnionFind uf = new UnionFind(N);
		int ans = 0;
		final int MOD = 1_000_000_007;
		int lastZ = -1;
		ArrayList<ArrayList<Edge>> line = new ArrayList<>(N);
		for (int i = 0;i < N;++ i) line.add(new ArrayList<>());
		for (int i = 0;i < M;++ i) {
			int x = ic.nextInt(1, N) - 1;
			ic.nextChar(' ');
			int y = ic.nextInt(1, N) - 1;
			if (x == y) throw new AssertionError();
			ic.nextChar(' ');
			int z = ic.nextInt(lastZ + 1, exponent10(1, 9));
			lastZ = z;
			ic.readNewLine();
			if (uf.isUnite(x, y)) continue;
			int weight = pow(X, z, MOD);
			line.get(x).add(new Edge(y, weight));
			line.get(y).add(new Edge(x, weight));
			uf.unite(x, y);
		}
		ic.checkEOF();
		if (uf.size(0) != N) throw new AssertionError();
		ArrayDeque<Integer> bfs = new ArrayDeque<>();
		for (int i = 0;i < N;++ i) if (line.get(i).size() == 1) bfs.add(i);
		int[] size = new int[N];
		boolean[] isArrived = new boolean[N];
		boolean[] isAdded = new boolean[N];
		Arrays.fill(size, 1);
		Arrays.fill(isArrived, false);
		Arrays.fill(isAdded, false);
		for (int i : bfs) isAdded[i] = true;
		while(!bfs.isEmpty()) {
			ArrayDeque<Integer> bfs2 = new ArrayDeque<>();
			while(!bfs.isEmpty()) {
				int tmp = bfs.poll();
				isArrived[tmp] = true;
				for (Edge e : line.get(tmp)) {
					if (isArrived[e.target]) continue;
					size[e.target] += size[tmp];
					if (!isAdded[e.target]) {
						bfs2.add(e.target);
						isAdded[e.target] = true;
					}
					ans = add(ans, mul(mul(size[tmp], N - size[tmp], MOD), e.weight, MOD), MOD);
				}
			}
			bfs = bfs2;
		}
		out.println(ans);
	}

	class Edge {
		final int target, weight;
		Edge(int target, int weight) {
			this.target = target;
			this.weight = weight;
		}
	}

	class UnionFind {
		final int[] parent;
		public UnionFind(int n) {
			parent = new int[n];
			java.util.Arrays.fill(parent, -1);
		}
		private int parent(int n) {
			if (parent[n] < 0) return n;
			return parent[n] = parent(parent[n]);
		}
		public int size(int n) {
			return -parent[parent(n)];
		}
		public boolean unite(int l, int r) {
			l = parent(l);
			r = parent(r);
			if (l == r) return false;
			if (parent[l] < parent[r]) {
				parent[l] += parent[r];
				parent[r] = l;
			} else {
				parent[r] += parent[l];
				parent[l] = r;
			}
			return true;
		}
		public boolean isUnite(int l, int r) {
			return parent(l) == parent(r);
		}
	}

	private int add(int a, int b, int mod) {
		return (a + b) % mod;
	}

	private int sub(int a, int b, int mod) {
		return (a - b + mod) % mod;
	}

	private int mul(int a, int b, int mod) {
		return (int)((long)a * b % mod);
	}

	private int div(int a, int b, int mod) {
		return mul(a, pow(b, mod - 2, mod), mod);
	}

	private int pow(int a, int b, int mod) {
		int ans = 1;
		for (int mul = a;b > 0;b >>= 1, mul = mul(mul, mul, mod)) if ((b & 1) != 0) ans = mul(ans, mul, mod);
		return ans;
	}

	public static int exponent10(int n, int e) {
		return n * pow(10, e);
	}

	public static long exponent10L(int n, int e) {
		return n * pow(10L, e);
	}

	public static int pow(int a, int b) {
		int ans = 1;
		for (int mul = a;b > 0;b >>= 1, mul *= mul) if ((b & 1) != 0) ans *= mul;
		return ans;
	}

	public static long pow(long a, long b) {
		long ans = 1;
		for (long mul = a;b > 0;b >>= 1, mul *= mul) if ((b & 1) != 0) ans *= mul;
		return ans;
	}

	public static class CharSet {
		private final BitSet set = new BitSet(256);

		public void add(char... c) {
			for (char i : c) set.set(i);
		}

		public void add(CharSet... s) {
			for (CharSet i : s) set.or(i.set);
		}

		public boolean contains(char... c) {
			for (char i : c) if (!set.get(i)) return false;
			return true;
	 	}

		public boolean contains(String s) {
			return contains(s.toCharArray());
	 	}

		private static final class Chars extends CharSet {
			public Chars(char... c) {
				super.add(c);
			}
			public Chars(CharSet... s) {
				super.add(s);
			}
			@Override
			public void add(char... c) {
				throw new UnsupportedOperationException();
			}
			@Override
			public void add(CharSet... s) {
				throw new UnsupportedOperationException();
			}
		}

		public static final CharSet NUMBER = new Chars('0','1','2','3','4','5','6','7','8','9');
		public static final CharSet LOWER = new Chars('a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z');
		public static final CharSet UPPER = new Chars('A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z');
		public static final CharSet ALPHABET = new Chars(LOWER, UPPER);

	}
	public static class InputChecker {
		private InputStream in;
		private final byte[] buffer = new byte[1024];
		private final byte[] undo = new byte[1024];
		private int undoSize = 0;
		private int read = 0;
		private int length = 0;

		public InputChecker(InputStream in) {
			this.in = in;
		}

		public final void setInputStream(InputStream in) {
			this.in = in;
		}

		public final void setInputStream(File in) {
			try {
				this.in = new FileInputStream(in);
			} catch (FileNotFoundException e) {
				e.printStackTrace();
			}
		}

		private boolean hasNextByte() {
			if (undoSize > 0) return true;
			if (read < length) return true;
			read = 0;
			try {
				length = in.read(buffer);
			} catch (IOException e) {
				e.printStackTrace();
			}
			return length > 0;
		}

		private byte readByte() {
			if (hasNextByte()) return undoSize > 0 ? undo[--undoSize] : buffer[read++];
			throw new NoSuchElementException();
		}

		private void undo(byte b) {
			undo[undoSize ++] = b;
		}

		private void undo(char c) {
			if ((c & 0xFF80) == 0) {
				undo((byte)c);
				return;
			}
			undo((byte)(c & 0x3F | 0x80));
			if ((c & 0xF800) == 0) {
				undo((byte)(c >> 6 & 0x1F | 0xC0));
			} else {
				undo((byte)(c >> 6 & 0x3F | 0x80));
				undo((byte)(c >> 12 | 0xE0));
			}
		}

		public final boolean hasNext() {
			return hasNextByte();
		}

		public final char nextChar() {
			byte b = readByte();
			if ((b & 0x80) == 0) return (char)b;
			if ((b & 0x20) == 0) return (char)((b & 0x1F) << 6 | (readByte() & 0x3F));
			return (char)((b & 0xF) << 12 | (readByte() & 0x3F) << 6 | (readByte() & 0x3F));
		}

		public final char nextChar(char estimate) {
			char c = nextChar();
			if (estimate == c) return estimate;
			undo(c);
			throw new AssertionError();
		}

		public final char nextChar(CharSet estimate) {
			char c = nextChar();
			if (estimate.contains(c)) return c;
			undo(c);
			throw new AssertionError();
		}

		public final void readNewLine() {
			char c = nextChar();
			if (c == '\r') {
				c = nextChar();
				if (c != '\n') undo(c);
				return;
			} else if (c == '\n') return;
			undo(c);
			throw new AssertionError();
		}

		public final void checkEOF() {
			if (hasNextByte()) throw new AssertionError();
		}

		public final String next(CharSet contains) {
			StringBuilder sb = new StringBuilder();
			try {
				do {
					char c = nextChar();
					if (!contains.contains(c)) {
						undo(c);
						return sb.toString();
					}
					sb.append(c);
				} while(true);
			} catch (NoSuchElementException e) {
				if (sb.length() <= 0) throw new AssertionError();
				return sb.toString();
			}
		}

		public final int nextInt() {
			byte b = readByte();
			int n = 0;
			if (b == '-') {
				if (!isNumber(b = readByte())) {
					undo(b);
					throw new NumberFormatException();
				}
				try {
					if (b == '0') {
						if (!isNumber(b = readByte())) {
							undo(b);
							return 0;
						}
						throw new NumberFormatException();
					}
					do n = Math.addExact(Math.multiplyExact(n, 10), '0' - b); while(isNumber(b = readByte()));
					undo(b);
				} catch (NoSuchElementException e) {
				}
				return n;
			}
			if (!isNumber(b)) {
				undo(b);
				throw new NumberFormatException();
			}
			try {
				if (b == '0') {
					if (!isNumber(b = readByte())) {
						undo(b);
						return 0;
					}
					throw new NumberFormatException();
				}
				do n = Math.addExact(Math.multiplyExact(n, 10), b - '0'); while(isNumber(b = readByte()));
				undo(b);
			} catch (NoSuchElementException e) {
			}
			return n;
		}

		public final int nextInt(int min, int max) {
			int n = nextInt();
			if (min <= n && n <= max) return n;
			throw new NumberFormatException();
		}

		private static boolean isNumber(byte c) {
			return '0' <= c && c <= '9';
		}

		public final long nextLong() {
			byte b = readByte();
			long n = 0;
			if (b == '-') {
				if (!isNumber(b = readByte())) {
					undo(b);
					throw new NumberFormatException();
				}
				try {
					if (b == '0') {
						if (!isNumber(b = readByte())) {
							undo(b);
							return 0;
						}
						throw new NumberFormatException();
					}
					do n = Math.addExact(Math.multiplyExact(n, 10), '0' - b); while(isNumber(b = readByte()));
					undo(b);
				} catch (NoSuchElementException e) {
				}
				return n;
			}
			if (!isNumber(b)) {
				undo(b);
				throw new NumberFormatException();
			}
			try {
				if (b == '0') {
					if (!isNumber(b = readByte())) {
						undo(b);
						return 0;
					}
					throw new NumberFormatException();
				}
				do n = Math.addExact(Math.multiplyExact(n, 10), b - '0'); while(isNumber(b = readByte()));
				undo(b);
			} catch (NoSuchElementException e) {
			}
			return n;
		}

		public final long nextLong(long min, long max) {
			long n = nextLong();
			if (min <= n && n <= max) return n;
			throw new NumberFormatException();
		}

		public final double nextDouble() {
			StringBuilder sb = new StringBuilder();
			byte b = readByte();
			if (b == '-') {
				sb.append(b);
				b = readByte();
			}
			if (b == '0') {
				sb.append(b);
				b = readByte();
			} else {
				while(isNumber(b)) {
					sb.append(b);
					b = readByte();
				}
			}
			if (b == '.') {
				sb.append(b);
				b = readByte();
				while(isNumber(b)) {
					sb.append(b);
					b = readByte();
				}
			}
			if (b == 'e' || b == 'E') {
				sb.append(b);
				b = readByte();
				if (b == '-' || b == '+') {
					sb.append(b);
					b = readByte();
				}
				while(isNumber(b)) {
					sb.append(b);
					b = readByte();
				}
			}
			undo(b);
			return Double.parseDouble(sb.toString());
		}
	}
}

yukicoder

結果

テストケース

ソースコード