828
技術社區[雲棲]
POJ 3370 鴿巢原理
題意:給出n個數 求出任意和 為c的倍數。有多組解輸出一組。
這題同POJ2356相似隻不過換了個問法,根據鴿巢原理實質上還是一定有解且是連續解。所以隻需要求出連續解就可以了。首先用s[1],s[2],s[3]...s[n]表示第一個數到第n個數的和也就是s[n]=a[1]+...+a[n],然後將s[i]分別對n取餘,顯然,如果s[i]=0那麼此時1-i個數即為n的倍數也就是答案,如果s[i]均不為0,那麼通過鴿巢原理可知s[i]個數為n,1<=s[i]<=n-1,那麼至少有兩個s[i] 即s[i]=s[j] (i!=j) 所以此時i+1至j 即為答案
#include <iostream> #include<cstdio> #include<cstring> using namespace std; int s[100005],bj[100005]; int main() { int c,n,a,f,ans1,ans2; while(~scanf("%d%d",&c,&n),c+n) { memset(bj,0,sizeof(bj)); s[0]=f=0; for(int i=1; i<=n; i++) { scanf("%d",&a); if(f) continue; s[i]=(s[i-1]+a)%c; if(s[i]==0) f=1,ans1=i; if(bj[s[i]]) f=2,ans1=bj[s[i]],ans2=i; else bj[s[i]]=i; } if(f==1) for(int i=1; i<=ans1; i++) if(i!=ans1) printf("%d ",i); else printf("%d\n",i); else for(int i=ans1+1; i<=ans2; i++) if(i!=ans2) printf("%d ",i); else printf("%d\n",i); } return 0; }
最後更新:2017-04-04 07:03:45