UOJ Logo

NOI.AC

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#202211#3533. 数组cqzxz100529ms13000kbC++2.3kb2024-02-14 10:06:422024-02-14 12:33:06

answer

#include<bits/stdc++.h>
#define rg register
#define il inline
#define ll long long
using namespace std;

const int INF = 0x7ffffff / 2, N = 100005;
ll ans[N << 3], res;
ll tag1[N << 3], tag2[N << 3];
int n, a[N];
vector<int> b[N];

il int rd(){
	char c = getchar(); 
	int n = 0;
	while (c < '0' || c > '9'){ 
		c = getchar(); 
	}
	while (c >= '0' && c <= '9'){ 
		n = (n << 1) + (n << 3) + (c & 15); 
		c = getchar(); 
	}
	return n;
}

il void pushup(int k){
	ans[k] = ans[k << 1] + ans[k << 1 | 1];
}

void modify(int k, ll v, ll d, int len){
	ans[k] += (v * 2 + (len - 1) * d) * len / 2;
	tag1[k] += v;
	tag2[k] += d;
}

void pushdown(int k, int len1, int len2){
	if (tag1[k] || tag2[k]){
		modify(k << 1, tag1[k], tag2[k], len1);
		modify(k << 1 | 1, tag1[k] + tag2[k] * len1, tag2[k], len2);
		tag1[k] = tag2[k] = 0;
	}
}

il void change(int k, int l, int r, int x, int y, int v, int d){
	if (l >= x && r <= y){
		modify(k, v + (l - x) * d, d, r - l + 1);
		return;
	}
	int mid = l + r >> 1;
	pushdown(k, mid - l + 1, r - mid);
	if (x <= mid) change(k << 1, l, mid, x, y, v, d);
	if (y >= mid + 1) change(k << 1 | 1, mid + 1, r, x, y, v, d);
	pushup(k);
}

il ll ask(int k, int l, int r, int x, int y){
	if (l >= x && r <= y) return ans[k];
	int mid = l + r >> 1;
	pushdown(k, mid - l + 1, r - mid);
	ll res = 0;
	if (x <= mid) res += ask(k << 1, l, mid, x, y);
	if (y >= mid + 1) res += ask(k << 1 | 1, mid + 1, r, x, y);
	return res;
}

signed main(){
	n = rd();
	for (int i = 1; i <= n; i++){
		a[i] = rd();
		b[a[i]].push_back(i);
	}
	int det = n + 1;
	for (int i = 0; i < n; i++){
		b[i].push_back(n + 1);
		int last = 0;
		for (int j = 0; j < b[i].size(); j++){
			int x = 2 * j - (b[i][j] - 1) + det, y = 2 * j - last + det;
			res += ask(1, 1, 2 * n + 1, max(x - 1, 1), y - 1);
			change(1, 1, 2 * n + 1, x, y, 1, 1);
			if (x + 1 <= 2 * n + 1) change(1, 1, 2 * n + 1, y + 1, 2 * n + 1, y - x + 1, 0);
			last = b[i][j];
		}
		last = 0;
		for (int j = 0; j < b[i].size(); j++){
			int x = 2 * j - (b[i][j] - 1) + det, y = 2 * j - last + det;
			change(1, 1, 2 * n + 1, x, y, -1, -1);
			if (x + 1 <= 2 * n + 1) change(1, 1, 2 * n + 1, y + 1, 2 * n + 1, -(y - x + 1), 0);
			last = b[i][j];
		}
	}
	cout << res;
	return 0;
}

详细

小提示:点击横条可展开更详细的信息

Test #1:

score: 10
Accepted
time: 0ms
memory: 3592kb

input:

100
1 1 1 1 1 8 1 1 8 1 4 2 2 2 2 2 2 2 2 8 3 3 5 3 3 6 3 3 3 3 4 4 4 4 2 4 4 4 4 4 5 5 2 9 5 5 7 9 ...

output:

844

result:

ok single line: '844'

Test #2:

score: 10
Accepted
time: 2ms
memory: 3684kb

input:

1000
1 1 1 8 1 1 1 1 5 1 1 9 1 4 8 8 1 1 1 1 1 1 1 10 1 1 1 1 9 1 1 1 1 1 1 1 1 10 1 1 1 1 1 1 2 1 1...

output:

95463

result:

ok single line: '95463'

Test #3:

score: 10
Accepted
time: 0ms
memory: 3680kb

input:

1000
1 1 1 10 5 8 1 1 1 1 6 7 1 1 1 1 1 1 1 1 1 6 9 1 1 1 9 1 1 1 5 9 9 3 1 4 3 1 1 1 1 3 10 1 4 7 4...

output:

88434

result:

ok single line: '88434'

Test #4:

score: 10
Accepted
time: 8ms
memory: 4708kb

input:

10000
1 3 1 1 1 1 1 6 1 1 1 1 7 5 1 1 1 1 1 1 3 3 3 1 1 1 1 7 1 1 4 1 1 8 1 9 1 1 1 7 1 1 1 1 1 2 1 ...

output:

9306221

result:

ok single line: '9306221'

Test #5:

score: 10
Accepted
time: 11ms
memory: 4720kb

input:

10000
1 1 1 1 9 1 10 1 1 1 1 1 1 1 3 1 1 1 1 8 1 1 1 10 1 1 1 1 1 1 5 1 6 1 8 1 3 9 5 1 5 3 1 1 3 1 ...

output:

9463489

result:

ok single line: '9463489'

Test #6:

score: 10
Accepted
time: 97ms
memory: 7444kb

input:

100000
2 2 2 2 1 1 2 2 2 1 2 1 1 2 2 2 2 1 1 1 1 1 1 2 2 2 2 2 1 2 1 2 2 1 2 1 2 2 1 1 1 2 2 2 2 2 1...

output:

4980632044

result:

ok single line: '4980632044'

Test #7:

score: 10
Accepted
time: 86ms
memory: 7452kb

input:

100000
1 2 1 1 1 2 1 1 1 2 1 1 2 1 2 2 1 1 2 2 1 2 1 2 2 2 2 2 1 1 1 1 1 1 1 1 2 1 2 2 1 2 1 1 1 2 1...

output:

4980856291

result:

ok single line: '4980856291'

Test #8:

score: 10
Accepted
time: 105ms
memory: 12348kb

input:

100000
1 1 1 2 1 1 1 1 1 1 1 2 2 1 1 2 1 1 1 1 1 1 1 1 2 2 1 2 1 1 2 2 1 1 2 1 1 1 1 1 1 1 1 1 1 1 1...

output:

992469861

result:

ok single line: '992469861'

Test #9:

score: 10
Accepted
time: 110ms
memory: 13000kb

input:

100000
1 1 2 1 5 3 1 1 1 2 1 2 10 10 1 1 1 1 1 1 3 1 1 1 3 1 1 8 1 1 1 1 1 1 4 1 1 1 1 5 1 1 1 1 6 8...

output:

963930312

result:

ok single line: '963930312'

Test #10:

score: 10
Accepted
time: 110ms
memory: 12992kb

input:

100000
6 1 1 6 1 1 1 1 2 1 5 4 1 6 1 10 1 1 1 1 1 2 7 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 3 1 1 2 1 8 1 ...

output:

963908847

result:

ok single line: '963908847'

Extra Test:

score: 0
Extra Test Passed