题意
有 n 个人做 m 道题,知道每个人做题的对错情况。每个人有一个预期得分 xi 和实际得分 ri。xi 给定,而 ri 是该人做对题目的分数之和。定义 “惊喜度” 为 每个人的∣xi−ri∣ 之和。请你构造出一个每道题的分数方案,使得:
- 分数是一个 1∼m 的排列。
- 在所有的方案中,该方案的 “惊喜度” 最大。
如果有多种可能的答案,你只要输出其中一组即可。
思路
学到了一个技巧,遇到绝对值的问题,可以尝试将绝对值分开,得到 ci * (xi - ri ) ,其中 ci 取 1 或 -1 ,由于 n <= 10 可以直接枚举出 ci 所有的可能,对于每一种可能,sum( ci * xi) - sum(ci *ri ),前一半直接得到,后一半中,ri 为该同学答对的题目乘该题目的分数,可以先计算出每一道题目总答对的次数,对该次数从小到大排序,根据排序不等式,出现次数越小的题目应分批越高的分数,通过枚举 ci 的每一种情况,维护最大值即可;
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[15];
string s[10010];
int ans[10010];
struct node{
int x,id;
}cnt[10010];
int main()
{
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
int _;
cin>>_;
while(_--){
int n,m,sum=-0x3f3f3f3f;
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++){
cin>>s[i];
s[i]=" "+s[i];
}
for(int i=0;i<(1ll<<n);i++){
for(int j=1;j<=m;j++){
cnt[j].x=0; cnt[j].id=j;
}
ll tmp=0;
for(int j=1;j<=n;j++){
if(i&(1<<(j-1))) tmp+=a[j];
else tmp-=a[j];
}
for(int j=1;j<=n;j++){
for(int k=1;k<=m;k++){
if(s[j][k]=='0') continue;
if(i&(1<<(j-1))) cnt[k].x++;
else cnt[k].x--;
}
}
sort(cnt+1,cnt+m+1,[](node a,node b){
return a.x>b.x;
});
for(int j=1;j<=m;j++) tmp-=j*cnt[j].x;
if(tmp>sum){
sum=tmp;
for(int j=1;j<=m;j++) ans[cnt[j].id]=j;
}
}
for(int i=1;i<=m;i++) cout<<ans[i]<<" ";
cout<<"\n";
}
return 0;
}