Orig Description
Score : 500 points
Problem StatementIn Takahashi Kingdom, which once existed, there are N cities, and some pairs of cities are connected bidirectionally by roads.
The following are known about the road network:
People traveled between cities only through roads. It was possible to reach any city from any other city, via intermediate cities if necessary.
Different roads may have had different lengths, but all the lengths were positive integers.
Snuke the archeologist found a table with N rows and N columns, A, in the ruin of Takahashi Kingdom.
He thought that it represented the shortest distances between the cities along the roads in the kingdom.
Determine whether there exists a road network such that for each u and v, the integer A_{u, v} at the u-th row and v-th column of A is equal to the length of the shortest path from City u to City v.
If such a network exist, find the shortest possible total length of the roads.
Constraints
1 \leq N \leq 300
If i ≠ j, 1 \leq A_{i, j} = A_{j, i} \leq 10^9.
A_{i, i} = 0
InputsInput is given from Standard Input in the following format:
N
A_{1, 1} A_{1, 2} ... A_{1, N}
A_{2, 1} A_{2, 2} ... A_{2, N}
...
A_{N, 1} A_{N, 2} ... A_{N, N}
OutputsIf there exists no network that satisfies the condition, print -1.
If it exists, print the shortest possible total length of the roads.
Sample Input 13
0 1 3
1 0 2
3 2 0
Sample Output 13
The network below satisfies the condition:
City 1 and City 2 is connected by a road of length 1.
City 2 and City 3 is connected by a road of length 2.
City 3 and City 1 is not connected by a road.
Sample Input 23
0 1 3
1 0 1
3 1 0
Sample Output 2-1
As there is a path of length 1 from City 1 to City 2 and City 2 to City 3, there is a path of length 2 from City 1 to City 3.
However, according to the table, the shortest distance between City 1 and City 3 must be 3.
Thus, we conclude that there exists no network that satisfies the condition.
Sample Input 35
0 21 18 11 28
21 0 13 10 26
18 13 0 23 13
11 10 23 0 17
28 26 13 17 0
Sample Output 382
Sample Input 43
0 1000000000 1000000000
1000000000 0 1000000000
1000000000 1000000000 0
Sample Output 43000000000
Extracted Specification
An integer N (1 ≤ N ≤ 300), representing some quantity or size.
A list of lists (2D matrix) A of size N x N, where each element A[i][j] is an integer:
- If i ≠ j: 1 ≤ A[i][j] = A[j][i] ≤ 10^9
- If i == j: A[i][j] = 0
### Example Input:
```
3
0 1 3
1 0 2
3 2 0
```
### Function Signature:
Write a function f(N, A) that takes in the input.
def f(N: int, A: List[List[int]]):
'''
N: an integer
A: a list of lists of integers
'''
Test Case Generator
Certainly! Let's write a test case generator for the provided problem statement.
We'll generate a matrix \( A \) according to the constraints, ensuring that:
1. \( A_{i, j} = A_{j, i} \) for \( i \neq j \) and \( A_{i, i} = 0 \).
2. \( 1 \leq A_{i, j} \leq 10^9 \) for \( i \neq j \).
Here's the test case generator in Python:
```python
import random
def generate_matrix(N):
A = [[0]*N for _ in range(N)]
for i in range(N):
for j in range(i+1, N):
distance = random.randint(1, 10**9)
A[i][j] = distance
A[j][i] = distance
return A
def tcgen():
N = random.randint(1, 300)
A = generate_matrix(N)
return N, A
def print_test_case(N, A):
print(N)
for row in A:
print(' '.join(map(str, row)))
# Example usage
N, A = tcgen()
print_test_case(N, A)
```
### Explanation:
1. **generate_matrix(N)**: This function generates a symmetric matrix \( A \) of size \( N \times N \) with the main diagonal set to 0 and other elements set to random positive integers between 1 and \( 10^9 \).
2. **tcgen()**: This function generates the size \( N \) of the matrix and the matrix \( A \).
3. **print_test_case(N, A)**: This function prints the generated test case in the required format.
You can run `tcgen()` to generate a new test case each time and `print_test_case(N, A)` to print the generated test case in the required format.
Feel free to adjust the random seed or add any specific conditions you want to test for.
Extract Arguments
def extract_arguments(fh):
N = int(fh.readline().strip())
A = []
for _ in range(N):
row = list(map(int, fh.readline().strip().split()))
A.append(row)
return N, A