Java实现蓝桥杯 历届试题 k倍区间

历届试题 k倍区间
时间限制：2.0s 内存限制：256.0MB
问题描述
　　给定一个长度为N的数列，A1, A2, … AN，如果其中一段连续的子序列Ai, Ai+1, … Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。

你能求出数列中总共有多少个K倍区间吗？
输入格式
　　第一行包含两个整数N和K。(1 <= N, K <= 100000)
　　以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)
输出格式
　　输出一个整数，代表K倍区间的数目。
样例输入
5 2
1
2
3
4
5
样例输出
6

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		int n, k;
		long ans = 0;
		Scanner sc = new Scanner(System.in);
		n=sc.nextInt();
		k=sc.nextInt();
		int [] sum = new int [100001];
		int [] num = new int [100001];
		int [] cnt = new int [100001];
		for(int i = 1; i <= n; i++){
			num[i] = sc.nextInt();
			sum[i] = (sum[i-1] + num[i])%k;
			ans += cnt[sum[i]];
			cnt[sum[i]]++;
		}
		long a = ans +(long)cnt[0];
System.out.println(a);
	}

}

import java.util.Scanner;


public class K倍区间 {
	 public static void main(String [] args) {
	        Scanner in = new Scanner(System.in);
	        int N = in.nextInt();
	        int k = in.nextInt();
	        int i,j;
	        int Ai[] = new int[N];
	        int count = 0;
	        int s = 0,t = 0;
	        for(int m=0;m<Ai.length;m++) {
	            Ai[m] = in.nextInt();
	        }
	        for( i=0;i<N;i++) {
	            s = s +Ai[i];
	                if(s%k == 0) {
	                    count ++;
	                }

	        }
	        System.out.println(count);
	        for(int m=0;m<N;m++) {
	            for(j=m+1;j<N;j++) {
	                t = t+Ai[j];
	                if(t%k == 0)
	                    count++;
	            }
	            t =0;
	            }
	        System.out.println(count);
	        }

}