牛客网 编程题 保卫方案

战争游戏的至关重要环节就要到来了，这次的结果将决定王国的生死存亡，小B负责首都的防卫工作。首都位于一个四面环山的盆地中，周围的n个小山构成一个环，作为预警措施，小B计划在每个小山上设置一个观察哨，日夜不停的瞭望周围发生的情况。 一旦发生外地入侵事件，山顶上的岗哨将点燃烽烟，若两个岗哨所在的山峰之间没有更高的山峰遮挡且两者之间有相连通路，则岗哨可以观察到另一个山峰上的烽烟是否点燃。由于小山处于环上，任意两个小山之间存在两个不同的连接通路。满足上述不遮挡的条件下，一座山峰上岗哨点燃的烽烟至少可以通过一条通路被另一端观察到。对于任意相邻的岗哨，一端的岗哨一定可以发现一端点燃的烽烟。 小B设计的这种保卫方案的一个重要特性是能够观测到对方烽烟的岗哨对的数量，她希望你能够帮她解决这个问题。
输入描述:
输入中有多组测试数据，每一组测试数据的第一行为一个整数n(3<=n<=10^6),为首都周围的小山数量，第二行为n个整数，依次表示为小山的高度h（1<=h<=10^9）.

输出描述:
对每组测试数据，在单独的一行中输出能相互观察到的岗哨的对数。

示例1
输入

5
1 2 4 5 3

输出

7

java代码如下，由于隔了一天，自动提交了，不知道AC结果，自己的几组数据通过测试了，有不对的地方欢迎指正，蟹蟹

import java.util.Scanner;

public class ProtectPlan {

    public static void main(String[] args){
        Scanner scan=new Scanner(System.in);
        int n=scan.nextInt();
        int[] h=new int[n];
        for(int i=0;i<n;i++){
            h[i]=scan.nextInt();
        }
        int num=sentryNum(n,h);
        System.out.println(num);
    }

    private static int sentryNum(int n, int[] h) {
        int num=0;
        int k=0;
        int l=0;
        boolean ite=true;;
        for(int i=0;i<n/2+1;i++){
            if(i<n-2){
                for(int j=i+2;j<n;j++){
                    ite=true;
                    if(i==0&&j==n-1){
                        //System.out.println("h["+i+"]和h["+j+"]为两个相邻的数");
                        break;
                    }
                    //System.out.println("h["+i+"]："+h[i]+"  h["+j+"]："+h[j]);
                    for(k=i+1;k<j;k++){
                        if(h[k]>h[i]||h[k]>h[j]){
                            //System.out.println("不联通的一对，顺时针："+h[i]+"和"+h[j]);
                            ite=false;
                            break;
                        }
                    }
                    if(ite){
                        //System.out.println("联通的一对，顺时针："+h[i]+"和"+h[j]);
                        num++;
                        continue;
                    }
                    if(i==0){
                        l=n;
                    }else{
                        l=i;
                    }
                    boolean cycle=false;
                    for(k=j+1;;k++){
                        if(k>=n){
                            k=k-n;
                            cycle=true;
                        }
                        if(k>=l&&cycle){
                            break;
                        }
                        ite=true;
                        if(h[k]>h[i]||h[k]>h[j]){
                            //System.out.println("不联通的一对，逆时针："+h[i]+"和"+h[j]);
                            ite=false;
                            break;
                        }
                    }
                    if(ite){
                        num++;
                        //System.out.println("联通的一对，逆时针："+h[i]+"和"+h[j]);
                    }
                }
            }
        }
        //System.out.println(num);
        num=n+num;
        return num;
    }
}