QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#609327#5507. InvestorsQHhee004WA 5ms4000kbC++172.0kb2024-10-04 12:08:072024-10-04 12:08:07

Judging History

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

  • [2024-10-04 12:08:07]
  • 评测
  • 测评结果:WA
  • 用时:5ms
  • 内存:4000kb
  • [2024-10-04 12:08:07]
  • 提交

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 clear() {
			for(int i=0; i<=n; i++)
				bit[i]=0;
		}
		void add_mx(int x,int t) {
			while(x)bit[x]=max(bit[x],t),x-=lowbit(x);
		}
		int get_mx(int x) {
			int ans=0;
			while(x<=n)ans=max(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.clear();
		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);
		}
	}
//	for(int L=1; L<=n; L++) {
//		for(int R=L; R<=n; R++) {
//			printf("[%d %d]=%d\n",L,R,f[L][R]);
//		}
//	}
}
struct node {
	int now,sum,cnt;
} ;
void Get_ans() {
	queue <node> Q;
	Q.push(node{1,0,0});
	int ans=inf;
	bit1.clear();
	while(!Q.empty()) {
		int x=Q.front().now,res=Q.front().sum,c=Q.front().cnt;
		Q.pop();
		if(c==m+1&&x==n+1)ans=min(ans,res);
		if(c==m+1||bit1.get_mx(x+1)>res)continue;
		bit2.add_mx(x,res);
		for(int i=x; i<=n; i++) {
			Q.push(node{i+1,res+f[x][i],c+1});
		}
		bit1.assign(bit2);
		bit2.clear();
	}
	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: 1ms
memory: 3808kb

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
Wrong Answer
time: 5ms
memory: 4000kb

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:

1
1000000007
1000000007
222
1
1000000007
1000000007
0
1000000007
1000000007
312
0
0
0
1
0
0
1000000007
1000000007
0
0
1000000007
1000000007
238
3
0
0
1
0
1000000007
1000000007
1000000007
1000000007
0
1
1000000007
3
0
0
1
0
1000000007
1
1000000007
1000000007
1
1000000007
1000000007
0
0
0
1000000007
1...

result:

wrong answer 2nd lines differ - expected: '18', found: '1000000007'