library/misc/fastio.hpp

• View this file on GitHub
• Last update: 2023-05-28 03:49:52+08:00
• Include: #include "library/misc/fastio.hpp"

Depends on

• library/misc/type-traits.hpp

Verified with

• test/convolution/ntt/yosupo-Convolution-Large.test.cpp
• test/misc/fastio/yosupo-Many-A+B-128-bit.test.cpp

Code

#pragma once
#include <iostream>
#include <cstring>
#include <type_traits>
#include "type-traits.hpp"

namespace std {

static struct FastInput {
	static constexpr int BUF_SIZE = 1 << 20;
	char buf[BUF_SIZE];
	size_t chars_read = 0;
	size_t buf_pos = 0;
	FILE *in = stdin;
	char cur = 0;

	inline char get_char() {
		if(buf_pos >= chars_read) {
			chars_read = fread(buf, 1, BUF_SIZE, in);
			buf_pos = 0;
			buf[0] = (chars_read == 0 ? -1 : buf[0]);
		}
		return cur = buf[buf_pos++];
		// return cur = getchar_unlocked();
	}

	inline void tie(int) {}

	inline explicit operator bool() { return cur != -1; }

	inline static bool is_blank(char c) { return c <= ' '; }

	inline bool skip_blanks() {
		while(is_blank(cur) && cur != -1) {
			get_char();
		}
		return cur != -1;
	}

	inline FastInput& operator>>(char& c) {
		skip_blanks();
		c = cur;
		return *this;
	}

	inline FastInput& operator>>(string& s) {
		if(skip_blanks()) {
			s.clear();
			do {
				s += cur;
			} while(!is_blank(get_char()));
		}
		return *this;
	}

	template<class T>
	inline FastInput& read_integer(T& n) {
		// unsafe, doesn't check that characters are actually digits
		n = 0;
		if(skip_blanks()) {
			int sign = +1;
			if(cur == '-') {
				sign = -1;
				get_char();
			}
			do {
				n += n + (n << 3) + cur - '0';
			} while(!is_blank(get_char()));
			n *= sign;
		}
		return *this;
	}

	template<class T>
	inline typename enable_if<felix::internal::is_integral<T>::value, FastInput&>::type operator>>(T& n) {
		return read_integer(n);
	}

	inline FastInput& operator>>(__int128& n) {
		return read_integer(n);
	}

	template<class T>
	inline typename enable_if<is_floating_point<T>::value, FastInput&>::type operator>>(T& n) {
		n = 0;
		if(skip_blanks()) {
			string s;
			(*this) >> s;
			sscanf(s.c_str(), "%lf", &n);
		}
		return *this;
	}
} fast_input;

#define istream FastInput
#define cin fast_input

static struct FastOutput {
	static constexpr int BUF_SIZE = 1 << 20;
	char buf[BUF_SIZE];
	size_t buf_pos = 0;
	static constexpr int TMP_SIZE = 1 << 20;
	char tmp[TMP_SIZE];
	FILE *out = stdout;
 
	inline void put_char(char c) {
		buf[buf_pos++] = c;
		if(buf_pos == BUF_SIZE) {
			fwrite(buf, 1, buf_pos, out);
			buf_pos = 0;
		}
		// putchar_unlocked(c);
	}

	~FastOutput() {
		fwrite(buf, 1, buf_pos, out);
	}

	inline FastOutput& operator<<(char c) {
		put_char(c);
		return *this;
	}

	inline FastOutput& operator<<(const char* s) {
		while(*s) {
			put_char(*s++);
		}
		return *this;
	}

	inline FastOutput& operator<<(const string& s) {
		for(int i = 0; i < (int) s.size(); i++) {
			put_char(s[i]);
		}
		return *this;
	}

	template<class T>
	inline char* integer_to_string(T n) {
		// beware of TMP_SIZE
		char* p = tmp + TMP_SIZE - 1;
		if(n == 0) {
			*--p = '0';
		} else {
			bool is_negative = false;
			if(n < 0) {
				is_negative = true;
				n = -n;
			}
			while(n > 0) {
				*--p = (char) ('0' + n % 10);
				n /= 10;
			}
			if(is_negative) {
				*--p = '-';
			}
		}
		return p;
	}

	template<class T>
	inline typename enable_if<felix::internal::is_integral<T>::value, char*>::type stringify(T n) {
		return integer_to_string(n);
	}

	inline char* stringify(__int128 n) {
		return integer_to_string(n);
	}

	template<class T>
	inline typename enable_if<is_floating_point<T>::value, char*>::type stringify(T n) {
		sprintf(tmp, "%.17f", n);
		return tmp;
	}

	template<class T>
	inline FastOutput& operator<<(const T& n) {
		auto p = stringify(n);
		for(; *p != 0; p++) {
			put_char(*p);
		}
		return *this;
	}
} fast_output;

#define ostream FastOutput
#define cout fast_output

} // namespace std

#line 2 "library/misc/fastio.hpp"
#include <iostream>
#include <cstring>
#include <type_traits>
#line 2 "library/misc/type-traits.hpp"
#include <cassert>

#include <numeric>

#line 5 "library/misc/type-traits.hpp"

namespace felix {

namespace internal {

#ifndef _MSC_VER
template<class T> using is_signed_int128 = typename std::conditional<std::is_same<T, __int128_t>::value || std::is_same<T, __int128>::value, std::true_type, std::false_type>::type;
template<class T> using is_unsigned_int128 = typename std::conditional<std::is_same<T, __uint128_t>::value || std::is_same<T, unsigned __int128>::value, std::true_type, std::false_type>::type;
template<class T> using make_unsigned_int128 = typename std::conditional<std::is_same<T, __int128_t>::value, __uint128_t, unsigned __int128>;
template<class T> using is_integral = typename std::conditional<std::is_integral<T>::value || is_signed_int128<T>::value || is_unsigned_int128<T>::value, std::true_type, std::false_type>::type;
template<class T> using is_signed_int = typename std::conditional<(is_integral<T>::value && std::is_signed<T>::value) || is_signed_int128<T>::value, std::true_type, std::false_type>::type;
template<class T> using is_unsigned_int = typename std::conditional<(is_integral<T>::value && std::is_unsigned<T>::value) || is_unsigned_int128<T>::value, std::true_type, std::false_type>::type;
template<class T> using to_unsigned = typename std::conditional<is_signed_int128<T>::value, make_unsigned_int128<T>, typename std::conditional<std::is_signed<T>::value, std::make_unsigned<T>, std::common_type<T>>::type>::type;
#else
template<class T> using is_integral = typename std::is_integral<T>;
template<class T> using is_signed_int = typename std::conditional<is_integral<T>::value && std::is_signed<T>::value, std::true_type, std::false_type>::type;
template<class T> using is_unsigned_int = typename std::conditional<is_integral<T>::value && std::is_unsigned<T>::value, std::true_type, std::false_type>::type;
template<class T> using to_unsigned = typename std::conditional<is_signed_int<T>::value, std::make_unsigned<T>, std::common_type<T>>::type;
#endif

template<class T> using is_signed_int_t = std::enable_if_t<is_signed_int<T>::value>;
template<class T> using is_unsigned_int_t = std::enable_if_t<is_unsigned_int<T>::value>;
template<class T> using to_unsigned_t = typename to_unsigned<T>::type;

template<class T> struct safely_multipliable {};
template<> struct safely_multipliable<short> { using type = int; };
template<> struct safely_multipliable<unsigned short> { using type = unsigned int; };
template<> struct safely_multipliable<int> { using type = long long; };
template<> struct safely_multipliable<unsigned int> { using type = unsigned long long; };
template<> struct safely_multipliable<long long> { using type = __int128; };
template<> struct safely_multipliable<unsigned long long> { using type = __uint128_t; };

template<class T> using safely_multipliable_t = typename safely_multipliable<T>::type;

}  // namespace internal


}  // namespace felix

#line 6 "library/misc/fastio.hpp"

namespace std {

static struct FastInput {
	static constexpr int BUF_SIZE = 1 << 20;
	char buf[BUF_SIZE];
	size_t chars_read = 0;
	size_t buf_pos = 0;
	FILE *in = stdin;
	char cur = 0;

	inline char get_char() {
		if(buf_pos >= chars_read) {
			chars_read = fread(buf, 1, BUF_SIZE, in);
			buf_pos = 0;
			buf[0] = (chars_read == 0 ? -1 : buf[0]);
		}
		return cur = buf[buf_pos++];
		// return cur = getchar_unlocked();
	}

	inline void tie(int) {}

	inline explicit operator bool() { return cur != -1; }

	inline static bool is_blank(char c) { return c <= ' '; }

	inline bool skip_blanks() {
		while(is_blank(cur) && cur != -1) {
			get_char();
		}
		return cur != -1;
	}

	inline FastInput& operator>>(char& c) {
		skip_blanks();
		c = cur;
		return *this;
	}

	inline FastInput& operator>>(string& s) {
		if(skip_blanks()) {
			s.clear();
			do {
				s += cur;
			} while(!is_blank(get_char()));
		}
		return *this;
	}

	template<class T>
	inline FastInput& read_integer(T& n) {
		// unsafe, doesn't check that characters are actually digits
		n = 0;
		if(skip_blanks()) {
			int sign = +1;
			if(cur == '-') {
				sign = -1;
				get_char();
			}
			do {
				n += n + (n << 3) + cur - '0';
			} while(!is_blank(get_char()));
			n *= sign;
		}
		return *this;
	}

	template<class T>
	inline typename enable_if<felix::internal::is_integral<T>::value, FastInput&>::type operator>>(T& n) {
		return read_integer(n);
	}

	inline FastInput& operator>>(__int128& n) {
		return read_integer(n);
	}

	template<class T>
	inline typename enable_if<is_floating_point<T>::value, FastInput&>::type operator>>(T& n) {
		n = 0;
		if(skip_blanks()) {
			string s;
			(*this) >> s;
			sscanf(s.c_str(), "%lf", &n);
		}
		return *this;
	}
} fast_input;

#define istream FastInput
#define cin fast_input

static struct FastOutput {
	static constexpr int BUF_SIZE = 1 << 20;
	char buf[BUF_SIZE];
	size_t buf_pos = 0;
	static constexpr int TMP_SIZE = 1 << 20;
	char tmp[TMP_SIZE];
	FILE *out = stdout;
 
	inline void put_char(char c) {
		buf[buf_pos++] = c;
		if(buf_pos == BUF_SIZE) {
			fwrite(buf, 1, buf_pos, out);
			buf_pos = 0;
		}
		// putchar_unlocked(c);
	}

	~FastOutput() {
		fwrite(buf, 1, buf_pos, out);
	}

	inline FastOutput& operator<<(char c) {
		put_char(c);
		return *this;
	}

	inline FastOutput& operator<<(const char* s) {
		while(*s) {
			put_char(*s++);
		}
		return *this;
	}

	inline FastOutput& operator<<(const string& s) {
		for(int i = 0; i < (int) s.size(); i++) {
			put_char(s[i]);
		}
		return *this;
	}

	template<class T>
	inline char* integer_to_string(T n) {
		// beware of TMP_SIZE
		char* p = tmp + TMP_SIZE - 1;
		if(n == 0) {
			*--p = '0';
		} else {
			bool is_negative = false;
			if(n < 0) {
				is_negative = true;
				n = -n;
			}
			while(n > 0) {
				*--p = (char) ('0' + n % 10);
				n /= 10;
			}
			if(is_negative) {
				*--p = '-';
			}
		}
		return p;
	}

	template<class T>
	inline typename enable_if<felix::internal::is_integral<T>::value, char*>::type stringify(T n) {
		return integer_to_string(n);
	}

	inline char* stringify(__int128 n) {
		return integer_to_string(n);
	}

	template<class T>
	inline typename enable_if<is_floating_point<T>::value, char*>::type stringify(T n) {
		sprintf(tmp, "%.17f", n);
		return tmp;
	}

	template<class T>
	inline FastOutput& operator<<(const T& n) {
		auto p = stringify(n);
		for(; *p != 0; p++) {
			put_char(*p);
		}
		return *this;
	}
} fast_output;

#define ostream FastOutput
#define cout fast_output

} // namespace std

Back to top page