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; }