题目链接
思路:我们枚举每一列,看看把每一列的总和变成最后一列总和相等的删除最小删除数量是多少,最后取数量最少的列就行,这里有个贪心思想,怎样删除才能使对方下降的快,我自己下降的少呢?其实和他们的距离差有关系,我们优先删除距离差较大的那一行。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e3+10;
typedef long long ll;
struct node{
int l,r,d,id;
};
int n,m,a[maxn][maxn];
vector<node>v[maxn];
vector<int>ans[maxn];
ll sum[maxn];
bool cmp(const node &a,const node &b)
{
return a.d==b.d?a.l<b.l:a.d>b.d;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;++i)for(int j=1;j<=n;++j)scanf("%d",&a[i][j]),sum[j]+=a[i][j];
for(int i=1;i<=m;++i)
for(int j=1;j<n;++j)
v[j].push_back({a[i][j],a[i][n],a[i][n]-a[i][j],i});
for(int j=1;j<n;++j)
sort(v[j].begin(),v[j].end(),cmp);
for(int j=1;j<n;++j)
{
int size=v[j].size(),k=0,sum1=sum[j],sum2=sum[n];
while(k<size)
{
if(sum1>=sum2) break;
sum1-=v[j][k].l,sum2-=v[j][k].r;
ans[j].push_back(v[j][k].id);
k++;
}
}
int minn=1e4+1,p;
for(int j=1;j<n;++j) {
if(ans[j].size()<minn) minn=ans[j].size(),p=j;
}
printf("%d\n",minn);
for(int i=0;i<ans[p].size();++i) printf("%d ",ans[p][i]);
}