CF1622E Math Test
卧槽,EDU 真难。😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭😭
考虑 \(n\le 10\) 实际上你可以直接枚举绝对值怎么拆,然后再求最值即可。
#include<bits/stdc++.h>
using namespace std;
const int N=11,M=1e4+5;
int n,m,p[M];long long res;
struct Student{int p,q[M];}a[N];
struct Problem{int cnt,id;}b[M];
bool cmp(Problem a,Problem b){return a.cnt<b.cnt;}
int solve(){
cin>>n>>m,res=-1;
for(int i=1;i<=n;++i) scanf("%d",&a[i].p);
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j)
scanf("%1d",&a[i].q[j]);
}
for(int i=0;i<(1<<n);++i){
for(int j=1;j<=m;++j){
b[j].id=j,b[j].cnt=0;for(int k=0;k<n;++k)
b[j].cnt+=a[k+1].q[j]*(((i>>k)&1)?1:-1);
}
sort(b+1,b+1+m,cmp);long long sum=0;
for(int j=0;j<n;++j) sum+=a[j+1].p*(((i>>j)&1)?-1:1);
// printf("%lld\n",sum);
// for(int j=1;j<=m;++j) printf("%d %d\n",b[j].id,b[j].cnt);
for(int j=1;j<=m;++j) sum+=b[j].cnt*j;
if(sum>res){res=sum;for(int j=1;j<=m;++j) p[b[j].id]=j;}
}
for(int i=1;i<=m;++i) printf("%d ",p[i]);
return printf("\n"),0;
}
int main(){
int T;cin>>T;while(T--) solve();
}
/*
8 4
2 7 7 9 4 1 6 10
0011
0100
0010
1111
1011
1111
0001
1010
*/