http://codeforces.com/problemset/problem/303/C
大致题意:给定一串数字,在最多移除k个数字的前提下,求出最小的数字m,使得不等式 恒成立。
分析:最简单的思路,遍历1-n,对于每个数字a[i]取模,设余数dex, cnt[dex]++,如果cnt[dex]>1 ans++; 如果ans<=k OK
然而,超时。。。 (巧妙的设置下标——用数组元素的计算结果作为dex,cnt[dex]++,统计个数在很多地方都能加速)
解决方法:a排序后,记录元素的“距离”——差的绝对值dex,edge[dex]++。这里有一个事实,如果要删除k个点,还有数字,至少有一个数字(它不会删 除所有的点吧!!),那么至少有1个点受到影响。这时的边最少,即k+1个点组成边的个数: k+(k-1)+(k-2)+……+1=(k+1)K/2.
借助于这个edge数组来加速判断:(风险最大安全最小——最大最小原理)
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=5e3+10,M=1e6+5;
int n,k;
int a[N],edge[M],cnt[M];
bool check(int key){
int ans=0;
for(int i=0;i<n;i++){
if(cnt[a[i]%key]==key){
ans++;
if(ans>k) return 0;
}
cnt[a[i]%key]=key;
}
return 1;
}
int main()
{
//freopen("cin.txt","r",stdin);
while(cin>>n>>k){
memset(edge,0,sizeof(edge));
memset(cnt,0,sizeof(cnt));
int Max=0;
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
Max=max(Max,a[i]);
}
sort(a,a+n);
for(int i=0;i<n;i++){
for(int j=i+1;j<n;j++){
int dex=a[j]-a[i];
edge[dex]++; //下标压缩 二元关系 用于加速
}
}
int ans=1,OK=k*(k+1)/2; //在二元关系下的k个点受影响的最少边数
for(int i=1;i<=Max+1;i++){
int cnt=0;
for(int p=i;p<=Max;p+=i){
cnt+=edge[p];
if(cnt>OK) break;
}
if(cnt<=OK&&check(i)){ ans=i; break; }
}
printf("%d\n",ans);
}
return 0;
}
再来一种写法:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
const int N=5005,M=1e6+10;
int n,k;
int a[N],dis[M];
int vis[M];
bool check(int val){
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++){
int dex=a[i]%val;
vis[dex]++;
}
int sum=0;
for(int i=0;i<val;i++){
if(vis[i]>1){
sum+=vis[i]-1;
if(sum>k) return 0;
}
}
return 1;
}
int main()
{
//freopen("cin.txt","r",stdin);
while(cin>>n>>k){
int Max=0;
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
Max=max(Max,a[i]);
}
sort(a,a+n);
memset(dis,0,sizeof(dis));
for(int i=0;i<n;i++){
for(int j=i+1;j<n;j++){
int dex=a[j]-a[i];
dis[dex]++;
}
}
int ans=0;
for(int i=1;i<=Max+1;i++){
int sum=0;
bool OK=1;
for(int j=i;j<=Max;j+=i){
sum+=dis[j];
if(sum>k*(k+1)/2){
OK=0;
break;
}
}
if(OK&&check(i)){ ans=i; break; }
}
printf("%d\n",ans);
}
return 0;
}