QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#609388#5507. InvestorsQHhee004ML 0ms5868kbC++172.1kb2024-10-04 12:41:302024-10-04 12:41:30

Judging History

你现在查看的是最新测评结果

  • [2024-10-04 12:41:30]
  • 评测
  • 测评结果:ML
  • 用时:0ms
  • 内存:5868kb
  • [2024-10-04 12:41:30]
  • 提交

answer

#include<iostream>
#include<map>
#include<algorithm>
#include<set>
#include<queue>
using namespace std;
const int inf=1e9+7,N=6005;
int n,m;
int tot,a[N],b[N],f[N][N];
#define lowbit(x) x&-x
class Tree_Array {
	private:
		int bit[N];
	public:
		void assign(Tree_Array &t) {
			for(int i=1; i<=n; i++)
				bit[i]=t.bit[i];
		}
		void set(int t) {
			for(int i=0; i<=n; i++)
				bit[i]=t;
		}
		void add_mn(int x,int t) {
			while(x)bit[x]=min(bit[x],t),x-=lowbit(x);
		}
		int get_mn(int x) {
			int ans=inf;
			while(x<=n)ans=min(bit[x],ans),x+=lowbit(x);
			return ans;
		}
		void add(int x,int y) {
			while(x)bit[x]+=y,x-=lowbit(x);
		}
		int get(int x) {
			int sum=0;
			while(x<=n)sum+=bit[x],x+=lowbit(x);
			return sum;
		}
} bit1,bit2;
void prepare() {
	scanf("%d%d",&n,&m);
	for(int i=1; i<=n; i++) {
		scanf("%d",&a[i]);b[i]=a[i];
	}
	map<int,int>F;
	tot=0;
	sort(b+1,b+1+n);
	for(int i=1; i<=n; i++)
		if(F[b[i]]==0)F[b[i]]=++tot;
	for(int i=1; i<=n; i++)a[i]=F[a[i]];
}
void Get_f() {
	for(int L=1; L<=n; L++) {
		bit1.set(0);
		bit1.add(a[L],1);
		for(int R=L+1; R<=n; R++) {
			f[L][R]=f[L][R-1]+bit1.get(a[R]+1);
			bit1.add(a[R],1);
		}
	}
}
struct node {
	int now,sum,cnt;
} ;
	queue <node> Q;
void Get_ans() {
	Q.push(node{1,0,0});
	int ans=inf;
	bit1.set(inf);
	bit2.set(inf);
	while(!Q.empty()) {
		int x=Q.front().now,res=Q.front().sum,c=Q.front().cnt;
		Q.pop();
//		if(c==10)printf("c:%d\n",c);
//		printf("[%d %d %d]\n",x,res,c);
		if(c==m+1&&x==n+1)ans=min(ans,res);
		if(c==m+1||x==n+1||res>=bit1.get_mn(x+1))continue;
		bit2.add_mn(x,res);
		for(int i=x; i<=n; i++) {
//			if(i<n&&res+f[x][i]>=bit2.get_mn(i+1))continue;
			Q.push(node{i+1,res+f[x][i],c+1});
		}
		if(Q.front().cnt>c) {
			bit1.assign(bit2);
			bit2.set(inf);
		}
	}
	printf("%d\n",ans);
}
void solve() {
	prepare();
	Get_f();
	Get_ans();
}
int main() {
//	freopen("data.in","r",stdin);
//	freopen("my.out","w",stdout);
	int T;scanf("%d",&T);
	while(T--) solve();
//	fclose(stdin);
//	fclose(stdout);
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 5868kb

input:

2
6 1
4 5 6 2 2 1
6 2
4 5 6 2 2 1

output:

2
0

result:

ok 2 lines

Test #2:

score: -100
Memory Limit Exceeded

input:

349
6 2
2 1 2 1 2 1
48 12
42 47 39 39 27 25 22 44 45 13 5 48 38 4 37 6 24 10 42 38 12 37 31 19 44 6 29 17 7 12 7 26 35 24 15 9 37 3 27 21 33 29 34 20 14 30 31 21
48 12
1 43 17 46 17 39 40 22 25 2 22 12 4 11 38 12 4 11 1 5 39 44 37 10 19 20 42 45 2 45 37 20 48 34 16 41 23 18 13 44 47 21 29 4 23 18 16...

output:


result: