2014 Super Training #7 B Continuous Login --二分

原题：ZOJ 3768 http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3768

一个结论：一个正整数总能用不超过三个前n项相加表示。

先找一个的，在找两个，三个的，二分找，用lower_bound函数。

代码：

#include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <cstdlib> #include <algorithm> #include <vector>

using namespace std; #define N 16007



int sum[N]; void init() { sum[0] = 0; for(int i=1;i<16000;i++) sum[i] = sum[i-1] + i; } int main() { int t,n,i,j; init(); int a,b; int x,y,z; scanf("%d",&t); while(t--) { scanf("%d",&n); int flag = 0; int t = lower_bound(sum,sum+16000,n)-sum; if(sum[t] == n) { printf("%d\n",t); continue; } for(i=t;i&&2*sum[i]>=n;i--) { if(flag == 2) break; int lef = n-sum[i]; int t2 = lower_bound(sum,sum+16000,lef)-sum; for(j=t2;j&&sum[i]+sum[j]*2>=n;j--) { if(flag == 2) break; if(sum[i]+sum[j]==n) { flag = 2; a = i,b = j; break; } else if(flag == 3) break; else if(sum[i]+sum[j]<n) { int lef2 = n-sum[i]-sum[j]; int t3 = lower_bound(sum,sum+16000,lef2)-sum; if(sum[t3] == lef2) { flag = 3; x = i,y = j,z = t3; break; } } } } if(flag == 2) printf("%d %d\n",a,b); else if(flag == 3) printf("%d %d %d\n",x,y,z); } return 0; }

View Code