-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathF.cpp
85 lines (75 loc) · 1.93 KB
/
F.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#include <algorithm>
#include <iostream>
#include <cstring>
#include <climits>
#include <cstdlib>
#include <cstdio>
#include <bitset>
#include <vector>
#include <cmath>
#include <ctime>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define fi first
#define se second
#define U unsigned
#define P std::pair
#define Re register
#define LL long long
#define pb push_back
#define MP std::make_pair
#define all(x) x.begin(),x.end()
#define CLR(i,a) memset(i,a,sizeof(i))
#define FOR(i,a,b) for(Re int i = a;i <= b;++i)
#define ROF(i,a,b) for(Re int i = a;i >= b;--i)
#define DEBUG(x) std::cerr << #x << '=' << x << std::endl
const int MAXN = 5e5 + 10;
const int ha = 1e9 + 7;
int n;
struct BIT{
#define lowbit(x) ((x)&(-x))
LL tree[MAXN];
inline void clear(){CLR(tree,0);}
inline void add(int pos,int x){
while(pos <= n){
(tree[pos] += x) %= ha;
pos += lowbit(pos);
}
}
inline LL calc(int pos){
LL res = 0ll;
while(pos){
(res += tree[pos]) %= ha;
pos -= lowbit(pos);
}
return res;
}
}bit;
int a[MAXN],origin[MAXN],cn;
std::map<int,int> S;
LL ans = 0ll;
inline void calc(int k){
if(!k) std::reverse(a+1,a+n+1);bit.clear();
FOR(i,1,n){
//bit.add(a[i],i); 写在这里会导致被算两遍
int t = bit.calc(a[i])%ha;
//printf("t:%d a[%d]:%d origin[a[%d]]:%d (n-i+1):%d\n",t,i,a[i],i,origin[a[i]],n-i+1);
(ans += 1ll*(n-i+1)*origin[a[i]]%ha*t%ha) %= ha;
if(k) (ans += 1ll*(n-i+1)*origin[a[i]]%ha*i%ha) %= ha;
bit.add(a[i],i);
}
}
int main(){
scanf("%d",&n);
FOR(i,1,n) scanf("%d",a+i),S.insert(MP(a[i],0));
for(auto &x:S){
x.second = ++cn;
origin[cn] = x.first;
}
FOR(i,1,n) a[i] = S[a[i]];
calc(1);calc(0);
printf("%lld\n",ans);
return 0;
}