USACO Silver 2019 February - Sleepy Cow Herding

C++

Copy
#include <bits/stdc++.h>
using namespace std;

int main() {
	freopen("herding.in", "r", stdin);

	int n;
	cin >> n;
	vector<int> herd(n);
	for (int i = 0; i < n; ++i) { cin >> herd[i]; }

Java

Copy
import java.io.*;
import java.util.*;

public class Herding {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new FileReader("herding.in"));

		int n = Integer.parseInt(br.readLine());
		int[] herd = new int[n];
		for (int i = 0; i < n; i++) { herd[i] = Integer.parseInt(br.readLine()); }

Python

Copy
with open("herding.in") as read:
	n = int(read.readline())
	herd = sorted(int(read.readline()) for _ in range(n))

min_moves = float("inf")
if herd[n - 2] - herd[0] == n - 2 and herd[n - 1] - herd[n - 2] > 2:
	min_moves = 2
elif herd[n - 1] - herd[1] == n - 2 and herd[1] - herd[0] > 2:
	min_moves = 2
else: