题意:给在n*m的网格里,每行选取一个数,使得最后选取的n个数之间的差值最小。
该题我一开始以为dp,但是弄了好久没有思路。后来在大神的指点下,才知道是个贪心。
假设最优解是a1,a2,a3;我们如果把所有数排好序放在一列,他们定会靠在一起,虽然也许不会一个仅挨一个。。。。
因为最终我们是想的到相差最小的来自不同行的数,我们可以先记录每个数出现在哪一行,然后把n*m个数排个序,
然后用一个两个指针(变量head,tail)从左往右去试探所有可行的区间(tail与head之间所有数刚好在以前每一行的都有),然后直接用tail-head的到一个解,然后head,tail前移,找到下一个合法区间,更新最小值ans。遍历时间复杂度2*(m*N),总时间n*M*log(n*M)。
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
struct node
{
int num;
int x;
bool operator <(const struct node &a) const
{
return x<a.x;
}
}a[10010],tmp;
//bool cmp(struct node b,struct node c)
//{
// return b.x<c.x;
//}
int v[110];
int main()
{
int n,m,i,ans,cnt,j,pos,head,tail,no,man;
while(scanf("%d%d",&n,&m)!=EOF)
{
//priority_queue<struct node,vector<struct node> ,cmp> que;
memset(v,0,sizeof(v));
ans=1e9;
pos=0;
for(i=0;i<n;i++)
{
for(j=0;j<m;j++){
a[pos].num=i;
scanf("%d",&a[pos++].x);
}
}
sort(a,a+pos);
// for(i=0;i<pos;i++)
// {
// printf("%d ",a[i].x);
// que.push(a[i]);
// }
// while(!que.empty())
// {
// printf("\n%d",que.top().x);
// que.pop();
// }
// system("pause");
head=tail=0;
cnt=0;
while(1)
{
while(tail<pos&&cnt<n)
{
tail++;
//printf("!%d %d\n",a[tail].x,man);
if(v[a[tail-1].num]==0)
cnt++;
v[a[tail-1].num]++;
}
while(cnt>=n)
{
ans=min(ans,a[tail-1].x-a[head].x);
if(v[a[head].num]==1)
{
cnt--;
}
v[a[head].num]--;
head++;
}
if(tail==pos)
{
break;
}
//printf("ans=%d %d\n",ans,tail);
}
printf("%d\n",ans);
}
}
以前知道贪心,但是实现太过暴力,TLE。以前的解法是先不用排序,直接在第一列找到最值,然后相减,得到一个解,然后去掉最小值,继续。。现在突然发现思路是错的。。。。。。。
tle code
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
int a[110][110];
int v[110];
int main()
{
int i,j,k,n,m,mini,maxi,ans,min_p;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(v,0,sizeof(v));
for(i=0;i<n;i++)
{
for(j=0;j<m;j++)
{
scanf("%d",&a[i][j]);
}
}
ans=100010;
while(1)
{
mini=1000000010;
maxi=-1;
for(i=0;i<n;i++)
{
if(mini>a[i][v[i]])
{
mini=a[i][v[i]];
min_p=i;
}
if(maxi<a[i][v[i]])
{
maxi=a[i][v[i]];
}
}
if(ans>maxi-mini)
{
ans=maxi-mini;
}
v[min_p]++;
//printf("%d %d %d %d\n",maxi,mini,min_p,v[min_p]);
//system("pause");
if(v[min_p]==m)
break;
}
printf("%d\n",ans);
}
}