Submission #7499891

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Studying https://atcoder.jp/contests/abc140/submissions/7391384
N = gets.to_i
P = gets.split.map &:to_i
Pi = [nil] * N
P.each_with_index{|p, i|
  Pi[p-1] = i
}
RU = (1..N).to_a + [N, N]
LU = (-1..(N-2)).to_a + [-1, -1]
p Pi.inject(0){|sum, i|
   l,  r = LU[i], RU[i]
  ll, rr = LU[l], RU[r]
  LU[r], RU[l] = l, r
  next sum + P[i] * ((i-l)*(rr-r) + (r-i)*(l-ll))
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

# Studying https://atcoder.jp/contests/abc140/submissions/7391384

N = gets.to_i
P = gets.split.map &:to_i

Pi = [nil] * N
P.each_with_index{|p, i|
	Pi[p-1] = i
}

RU = (1..N).to_a + [N, N]
LU = (-1..(N-2)).to_a + [-1, -1]

p Pi.inject(0){|sum, i|
	 l,  r = LU[i], RU[i]
	ll, rr = LU[l], RU[r]
	LU[r], RU[l] = l, r
	next sum + P[i] * ((i-l)*(rr-r) + (r-i)*(l-ll))
}

Submission Info

Compile Error

./Main.rb:4: warning: `&' interpreted as argument prefix

Judge Result