#775411 (D) No.1996 <><

提出ソース

結果

問題	No.1996 <><
ユーザー	daut-dlang
提出日時	2022-07-01 23:25:08
言語	D (dmd 2.107.1)
結果	AC
実行時間	1,059 ms / 2,000 ms
コード長	4,117 bytes
コンパイル時間	1,033 ms
コンパイル使用メモリ	116,508 KB
実行使用メモリ	6,784 KB
最終ジャッジ日時	2023-09-04 17:40:55
合計ジャッジ時間	12,694 ms
ジャッジサーバーID （参考情報）	judge12 / judge11

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,380 KB
testcase_01	AC	1 ms 4,380 KB
testcase_02	AC	1 ms 4,380 KB
testcase_03	AC	1 ms 4,380 KB
testcase_04	AC	1 ms 4,376 KB
testcase_05	AC	1 ms 4,380 KB
testcase_06	AC	1 ms 4,380 KB
testcase_07	AC	2 ms 4,384 KB
testcase_08	AC	3 ms 4,380 KB
testcase_09	AC	2 ms 4,376 KB
testcase_10	AC	3 ms 4,380 KB
testcase_11	AC	534 ms 6,732 KB
testcase_12	AC	1,059 ms 6,760 KB
testcase_13	AC	978 ms 6,700 KB
testcase_14	AC	947 ms 6,728 KB
testcase_15	AC	796 ms 6,716 KB
testcase_16	AC	682 ms 6,708 KB
testcase_17	AC	836 ms 6,704 KB
testcase_18	AC	453 ms 6,752 KB
testcase_19	AC	556 ms 6,748 KB
testcase_20	AC	546 ms 6,736 KB
testcase_21	AC	719 ms 6,784 KB
testcase_22	AC	841 ms 6,700 KB
testcase_23	AC	33 ms 6,744 KB
testcase_24	AC	471 ms 6,716 KB
testcase_25	AC	221 ms 6,700 KB
testcase_26	AC	391 ms 6,700 KB
testcase_27	AC	78 ms 6,728 KB
testcase_28	AC	1 ms 4,380 KB
testcase_29	AC	200 ms 6,744 KB
testcase_30	AC	11 ms 5,480 KB
testcase_31	AC	84 ms 6,716 KB
testcase_32	AC	4 ms 4,380 KB

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

ソースコード

raw source code

import std.stdio, std.conv, std.functional, std.string;
import std.algorithm, std.array, std.container, std.range, std.typecons;
import std.bigint, std.numeric, std.math, std.random;
import core.bitop;

string FMT_F = "%.10f";

static File _f;
void file_io(string fn) { _f = File(fn, "r"); }
static string[] s_rd;
T _RD(T = long)() { while(!s_rd.length) s_rd = readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T _RD(T = long)(File f) { while(!s_rd.length) s_rd = f.readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T[] _RDA(T = long)(T fix = 0) { auto r = readln.chomp.split.to!(T[]); r[] += fix; return r; }
T[] _RDA(T = long)(File f, T fix = 0) { auto r = f.readln.chomp.split.to!(T[]); r[] += fix; return r; }
T RD(T = long)() { if (_f.isOpen) return _RD!T(_f); else return _RD!T; }
T[] RDA(T = long)(T fix = 0) { if (_f.isOpen)  return _RDA!T(_f, fix); else return _RDA!T(fix); }

size_t[] MAKE_IDX(alias less = "a < b", Range)(Range range) { auto idx = new size_t[](range.length); makeIndex!(less)(range, idx); return idx;}
size_t MIN_POS(alias less = "a < b", Range)(Range range) { auto r = minPos!(less)(range); return range.length - r.length; }

void chmin(T)(ref T x, T y) { x = min(x, y); } void chmax(T)(ref T x, T y) { x = max(x, y); }
bool inside(T)(T x, T b, T e) { return x >= b && x < e; }
T lcm(T)(T x, T y) { return x * (y / gcd(x, y)); }
double euDist(T)(T[] a, T[] b) { auto c = a.dup; c[] -= b[]; c[] *= c[]; return sqrt(cast(double)c.sum); }
double[] rotate(double[] vec, double rad) { return [cos(rad)*vec[0] - sin(rad)*vec[1], sin(rad)*vec[0] + cos(rad)*vec[1]]; }
double norm(double[] vec) { return sqrt(reduce!((a,b)=>a+b*b)(0.0, vec)); }
double dotProd(double[] a, double[] b) { auto r = a.dup; r[] *= b[]; return r.sum; }

long mod = 10^^9 + 7;
//long mod = 998_244_353;
//long mod = 1_000_003;
void moda(ref long x, long y) { x = (x + y) % mod; }
void mods(ref long x, long y) { x = ((x + mod) - (y % mod)) % mod; }
void modm(ref long x, long y) { x = (x * y) % mod; }
void modpow(ref long x, long y) { if (!y) { x = 1; return; } auto t = x; x.modpow(y>>1); x.modm(x); if (y&1) x.modm(t); }
void modd(ref long x, long y) { y.modpow(mod - 2); x.modm(y); }

long compress(in long[] arr, out long[long] zip, out long[] unzip)
{
	auto index = MAKE_IDX(arr);
	long last = arr[index[0]], x;
	zip[last] = x;
	unzip ~= last;
	foreach (i; index[1..$])
	{
		if (arr[i] == last) continue;
		last = arr[i];
		++x;
		zip[last] = x;
		unzip ~= last;
	}
	return x+1;
}

struct SegTree(T)
{
	T delegate(const(T), const(T)) f;
	T[] data;
	T init;
    this(T[] _data, T delegate(const(T), const(T)) _f, T _init)
    {
		f = _f; init = _init;
		size_t n = 1; while (n < _data.length) n *= 2; data.length = n*2-1;
		foreach (i; 0.._data.length) data[i+n-1] = _data[i];
		foreach (i; _data.length..n) data[i+n-1] = _init;
		foreach_reverse (i; 0..n-1) data[i] = f(data[i*2+1], data[i*2+2]);
	}
    T query(size_t l, size_t r)
	{
        size_t n = (data.length+1) / 2; l += n-1; r += n-1; T res = init;
        while (l < r)
		{
            if ((l & 1) == 0) res = f(res, data[l]);
            if ((r & 1) == 0) res = f(res, data[r-1]);
            l = l/2; r = (r-1)/2;
        }
        return res;
    }
	void update(size_t i, T v)
	{
		i += (data.length+1) / 2 - 1; data[i] = v;
		while (i != 0) { i = (i-1)/2; data[i] = f(data[i*2+1], data[i*2+2]); }
    }
}

void main()
{
	auto N = RD;
	auto K = RD;
	auto A = RDA;
	auto s = RD!string;

	long[long] zip;
	long[] unzip;
	auto len = compress(A, zip, unzip);
	auto b = new long[](N);
	foreach (i; 0..N)
		b[i] = zip[A[i]];

	auto dp = new long[](N);
	dp[] = 1;
	foreach (i; 0..K)
	{
		auto ndp = new long[](N);
		auto data = new long[](N);
		auto st = SegTree!(long)(data, (a, b)=>(a+b)%mod, 0);
		foreach (j; 0..N)
		{
			if (s[i] == '<')
				ndp[j] = st.query(0, b[j]);
			else
				ndp[j] = st.query(b[j]+1, len);
			auto x = st.query(b[j], b[j]+1);
			x.moda(dp[j]);
			st.update(b[j], x);
		}
		dp = ndp;
	}

	long ans;
	foreach (e; dp)
		ans.moda(e);
	writeln(ans);
	stdout.flush;
	debug readln;
}

yukicoder

結果

テストケース

ソースコード