結果

問題 No.118 門松列(2)
ユーザー uafr_cs
提出日時 2015-09-02 17:44:15
言語 Java
(openjdk 23)
結果
AC  
実行時間 623 ms / 5,000 ms
コード長 1,279 bytes
コンパイル時間 2,175 ms
コンパイル使用メモリ 79,080 KB
実行使用メモリ 49,404 KB
最終ジャッジ日時 2024-10-09 07:06:18
合計ジャッジ時間 13,760 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeMap;

public class Main {
	
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		
		final int N = sc.nextInt();
		final long MOD = 1000000007;
		
		Map<Long, Integer> counts = new TreeMap<Long, Integer>();
		
		for(int i = 0; i < N; i++){
			final long L = sc.nextLong();
			
			if(!counts.containsKey(L)){
				counts.put(L, 0);
			}
			counts.put(L, counts.get(L) + 1);
		}
		//System.out.println(counts);
		
		long[] count_array = new long[counts.size()];
		long sum = 0;
		{
			int i = 0;
			for(final Entry<Long, Integer> entry : counts.entrySet()){
				count_array[i] = entry.getValue();
				sum += count_array[i];
				
				i++;
			}
		}
		
		long answer = 0;
		long forward_sum = 0, backward_sum = sum;
		for(int i = 0; i < count_array.length; i++){
			backward_sum -= count_array[i];
			
			//System.out.println(sum + " : " + forward_sum + " " + count_array[i] + " " + backward_sum);
			answer += forward_sum * count_array[i] * backward_sum;
			answer %= MOD;
			
			forward_sum += count_array[i];
		}
		
		System.out.println(answer);
		
	}
	
}
0