【Shellscript】AtCoder用g++コマンドを自作した話

#!/bin/bash

ACL_PATH="../ac-library" # ACLへのパス

# 使い方
function Usage() {
	cat 1>&2 <<EOF
Usage: ag++ [OPTIONS] <filename>

Options:
	-h	Display Usage(this message)
	-t	Compile in test mode
EOF
}

# オプション解析
while getopts ':th' OPT
do
	case $OPT in
		h)	# -hオプションは使い方紹介
			Usage
			exit 1 ;;
		t)	# -tオプションはテストモードとしてコンパイルする
			flag='test' ;;
		?)	# 不明なオプションについてエラー処理
			echo -e "ag++: \x1b[31merror:\x1b[0m illegal option. try -h to show usage."
			exit 1 ;;
	esac
done

shift $((OPTIND - 1)) # オプション解析後、切り詰める
arg=$@ # オプションを除いたコマンドライン引数

if [ $# -eq 0 ]; then # コマンドライン引数がない場合は-hの使用を促す
	echo -e "ag++: \x1b[31merror:\x1b[0m no input files. try -h to show usage."
	exit 1
fi
if [ -z ${flag} ]; then # -tオプションが指定されていない場合、本番用でコンパイル
	DEFINE="-DONLINE_JUDGE"
	echo -e "ag++: \x1b[36mCompile in submit mode\x1b[0m"
else # -tオプションが指定されている場合、テスト用でコンパイル
	echo -e "ag++: \x1b[36mCompile in test mode\x1b[0m"
fi
g++ $arg -std=gnu++17 $DEFINE -Wall -Wextra -O2 -I $ACL_PATH

shift $((OPTIND - 1)) # オプション解析後、切り詰める

#include <iostream>
#include <string>
using namespace std;

int main()
{
	int a, b, c;
	string s;
	cin >> a >> b >> c >> s;
#ifndef ONLINE_JUDGE
	cout << "a = " << a << endl;
	cout << "b = " << b << endl;
	cout << "c = " << c << endl;
	cout << "s = " << s << endl;
#endif
	cout << a + b + c << " " << s << endl;

	return 0;
}

$ ag++ -h
Usage: ag++ [OPTIONS] <filename>

Options:
        -h      Display Usage(this message)
        -t      Compile in test mode
$ ag++ PC_A.cpp
ag++: Compile in submit mode
$ ./a.out
1
2 3
test
6 test
$ ag++ -t PC_A.cpp
ag++: Compile in test mode
$ ./a.out
1
2 3
test
a = 1
b = 2
c = 3
s = test
6 test
$

#include <iostream>
#include <atcoder/all>

using namespace std;
using namespace atcoder;

int main()
{
	int N, Q;
	cin >> N >> Q;
	vector<int> t(Q), u(Q), v(Q);
	dsu tree(N);

	for (int i = 0; i < Q; i++) cin >> t[i] >> u[i] >> v[i];

	for (int i = 0; i < Q; i++){
		if (t[i]) cout << tree.same(u[i], v[i]) << endl;
		else tree.merge(u[i], v[i]);
	}

	return 0;
}

$ ag++ ALPC_A.cpp
ag++: Compile in submit mode
$ ./a.out
4 7
1 0 1
0 0 1
0 2 3
1 0 1
1 1 2
0 0 2
1 1 3
0
1
0
1
$