B. The Best Gift 傳送門:http://codeforces.com/problemset/problem/609/B
Emily's birthday is next week and Jack has decided to buy a present for her. He knows she loves books so he goes to the local bookshop, where there are n books on sale from one of m genres.
In the bookshop, Jack decides to buy two books of different genres.
Based on the genre of books on sale in the shop, find the number of options available to Jack for choosing two books of different genres for Emily. Options are considered different if they differ in at least one book.
The books are given by indices of their genres. The genres are numbered from 1 to m.
InputThe first line contains two positive integers n and m (2 ≤ n ≤ 2·105, 2 ≤ m ≤ 10) — the number of books in the bookstore and the number of genres.
The second line contains a sequence a1, a2, ..., an, where ai (1 ≤ ai ≤ m) equals the genre of the i-th book.
It is guaranteed that for each genre there is at least one book of that genre.
OutputPrint the only integer — the number of ways in which Jack can choose books.
It is guaranteed that the answer doesn't exceed the value 2·109.
Sample test(s) input4 3output
2 1 3 1
5input
7 4output
4 2 3 1 2 4 3
18Note
The answer to the first test sample equals 5 as Sasha can choose:
沒錯我超時了··· test19肯定是個大數據,然而一時想不出什麼方法來優化。占坑留著吧 日後再補
1 #include <stdio.h> 2 #include <stdlib.h> 3 int books[20000]; 4 int main(int argc, char *argv[]) 5 { 6 int n,m; 7 __int64 ways; 8 while(scanf("%d%d",&n,&m)!=EOF) 9 { 10 ways=0; 11 int i,j; 12 //for(i=0;i<n;i++) 13 //scanf("%d",&books[i]); 14 scanf("%d",&books[0]); 15 for(i=1;i<n;i++) 16 { 17 scanf("%d",&books[i]); 18 for(j=0;j<i;j++) 19 { 20 if(books[i]!=books[j]) 21 ways++; 22 } 23 } 24 printf("%I64d\n",ways); 25 } 26 return 0; 27 }
今天上cf看了看別人的代碼,用數學方法AC了···
#include <stdio.h> #include <stdlib.h> #include <memory.h> int a[200000],b[11];//b組用來記錄書的種類 int main(int argc, char *argv[]) { int n,m; int i,count; int k; while(scanf("%d%d",&n,&m)!=EOF) { memset(b,0,sizeof(0)); k=0; for(i=0;i<n;i++) { scanf("%d",&a[i]); b[a[i]]++; //相應種類的書籍數目++ } count=n; for(i=1;i<=m;i++) { count=count-b[i]; k+=b[i]*count;//b[i]種類書的數量乘上剩下其他種類書籍的數量 } printf("%d\n",k); } return 0; }