Funny Positive Sequence 找正序列

http://acm.fzu.edu.cn/problem.php?pid=1914

题目大意是序列A(a1,a2,a3......an),序列B(b1,b2,b3......bn)，且序列B由序列A生成(bi=a₁+a₂,+…+a_i (1≤i≤n))，若序列B内元素都为正数，则称序列A为一个正序列。

现在左移序列A内的元素0,1,2.....n-1次，产生n个新的序列：

A(0): a₁,a₂,…,a_n-1,a_n

A(1): a₂,a₃,…,a_n,a₁

…

A(n-2): a_n-1,a_n,…,a_n-3,a_n-2

A(n-1): a_n,a₁,…,a_n-2,a_n-1

问{ A(0), A(1), …, A(n-2), A(n-1) }内有多少个正序列。

总体思想是先找到一个<=0的数a_i，然后向前加(a_i+a_i-1)，若小于等于0说明该元素(a_i-1)不可能放在序列首（因为已经不满足和为正数的要求）

这样向前循环遍历一遍序列就可找出所有不可放在序列首的元素。剩下的元素即是正序列的个数。

 1 #include<stdio.h>
 2 #include<stdlib.h>
 3 #include<string.h>
 4 int p[500001], flag[500001];
 5 int main()
 6 {
 7     int m , n;
 8     int i, index, ans, cas = 0;
 9     long long sum;
10     scanf("%d", &m);
11     while(cas < m)
12     {
13               index = -1;
14               sum = 0;
15               memset(flag, 0, sizeof(flag));
16               scanf("%d", &n);
17               for (i = 0;i < n;i++)
18               {
19                       scanf("%d", &p[i]);
20                       if(p[i] <= 0)
21                               index = i;
22                       sum += p[i];
23               }
24               if (index == -1)
25               {//都为正数 ,直接输出n 
26                       printf("Case %d: %d\n", ++cas, n);
27                       continue;
28               }
29               if (sum <= 0)
30               {//和小于等于0,直接输出0 
31                       printf("Case %d: 0\n", ++cas);
32                       continue;
33               }
34               flag[index] = -1;
35               sum = p[index];
36               for (i = (index - 1 + n)%n; i != index; i = (i - 1 + n)%n)
37               {
38                     sum += p[i];
39                     if(sum > 0)
40                     {
41                            sum = 0;
42                            continue;
43                     }
44                     flag[i] = -1;    
45               }
46               i = index;
47               sum += p[i]; 
48               while (sum <= 0)
49               {
50                     flag[i] = -1;
51                     i = (i - 1 + n)%n;    
52                     sum += p[i];        
53               }  
54               ans = 0;
55               for (i = 0; i < n; i++)
56                     if (flag[i] == 0)
57                     ans++;
58               printf("Case %d: %d\n", ++cas, ans);
59     }
60     return 0;
61 }