Submission #1883346


Source Code Expand

#include<iostream>
#include <algorithm>
using namespace std;
struct node{
	long long num;
	int id;
	int numx;
}a[2000007];
node b[2000007];
long long tree[2000007];
inline int lowbit(int x) {
	return (x & -x);
}
void add(int x, int v) {
	for (int i = x; i <= 200007; i += lowbit(i)) {
		tree[i] += v;
	}
}
long long get(int x) {
	long long sum = 0;
	for (int i = x; i  ; i -= lowbit(i)) {
		sum += tree[i];
	}
	return sum;
}
int cmp(node a,node b){
	return a.num<b.num;
}
int cmp2(node a,node b){
	return a.id<b.id;
}
int main() {
	ios::sync_with_stdio(0);
	int n, k;
	cin >> n >> k;
	b[0].num=a[0].num=0;
	for(int i = 1; i <= n; i++) {
		cin >> a[i].num;
		a[i].num-=k;
		a[i].id=i;
		b[i].num=b[i-1].num+a[i].num;
		b[i].id=i;
	}
	b[0].id=0;
	sort(b,b+1+n,cmp);
	int t=1;
	b[0].numx=1;
	for (int i = 1; i <=n ; ++i)
	{	
		b[i].numx=b[i].num==b[i-1].num?t:++t;
	  
	}
	sort(b,b+1+n,cmp2);
	long long sum=0;
	for (int i = 0; i <=n ; ++i)
	{
	//cout<<a[i].num<<" "<<b[i].num<<" "<<b[i].numx<<endl;	
	}

	for(int i=n;i>=0;i--){

		
	//	cout<<i<<" "<<b[i].numx-1<<" "<<get(n+1)<<" "<<get(b[i].numx-1)<<endl;
		sum+=get(n+1)-get(b[i].numx-1);
		add(b[i].numx,1);
	}

	cout << sum << endl;
}

Submission Info

Submission Time
Task E - Meaningful Mean
User zqqian
Language C++14 (GCC 5.4.1)
Score 600
Code Size 1254 Byte
Status AC
Exec Time 78 ms
Memory 14080 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 600 / 600
Status
AC × 3
AC × 23
Set Name Test Cases
Sample a01, a02, a03
All a01, a02, a03, b04, b05, b06, b07, b08, b09, b10, b11, b12, b13, b14, b15, b16, b17, b18, b19, b20, b21, b22, b23
Case Name Status Exec Time Memory
a01 AC 2 ms 4352 KB
a02 AC 2 ms 4352 KB
a03 AC 2 ms 4352 KB
b04 AC 2 ms 4352 KB
b05 AC 53 ms 12544 KB
b06 AC 47 ms 14080 KB
b07 AC 78 ms 13312 KB
b08 AC 78 ms 13312 KB
b09 AC 66 ms 14080 KB
b10 AC 68 ms 14080 KB
b11 AC 2 ms 4352 KB
b12 AC 2 ms 4352 KB
b13 AC 24 ms 9088 KB
b14 AC 67 ms 14080 KB
b15 AC 41 ms 13312 KB
b16 AC 57 ms 14080 KB
b17 AC 66 ms 14080 KB
b18 AC 67 ms 14080 KB
b19 AC 48 ms 14080 KB
b20 AC 61 ms 12544 KB
b21 AC 52 ms 12544 KB
b22 AC 76 ms 14080 KB
b23 AC 75 ms 14080 KB